测试题目：两个有序数组，找出最大的五个数字组成一个新的数组

注意点：

1.输入两行的方法

2.两行输入的数量和小于5的情况

 

//评测题目: 两个有序数组，找出最大的五个数字组成一个新的数组
#include <iostream>
#include <vector> 
#include <cstring>
#include <bits/stdc++.h> 
using namespace std;

vector<int> getTop5(vector<int>& data1, vector<int>& data2, int& up){
    int i=data1.size()-1;
    int j=data2.size()-1;
    vector<int> res;
    for(int k=0; k<up;k++){
        if( i<0 || j<0 ) break;
        if(data1[i] >= data2[j] ){
            res.push_back( data1[i] );
            i--;
        }
        else{
            res.push_back( data2[j] );
            j--;            
        }
    }
    
    if(res.size()<up){  //注意有可能不够5个。
        if(i == -1 ){
            for(int m=j; res.size()<up && m>=0; m-- ){
                res.push_back( data2[m] ); 
            }
        }
        
        if(j == -1){
            for(int n=i; res.size()<up && n>=0; n-- ){
                res.push_back( data1[n] );
            }            
        } 
    }
    
    return res;
} 

int main(){
    
    int up = 5;
    vector<int> data1;
    vector<int> data2;
    string str; 
    bool sign1=0;  //代表data1已经读入 
    bool sign2=0;  //代表data2已经读入 
    while( getline(cin, str) ){
        if(!sign1 ){
            istringstream temp1(str);
            int cur1;
            while(temp1>>cur1){
                data1.push_back(cur1);
            }            
            sign1 = 1;
        }
        else if(!sign2 ){
            istringstream temp2(str);
            int cur2;
            while(temp2>>cur2){
                data2.push_back(cur2); 
            }
            sign2 = 1;
        }
        
        if(sign1&&sign2) {
                        
            vector<int> res = getTop5(data1, data2, up);
            if(res.size()<up) cout<<"nums are not enough\n";
            else{
                for(int i=0; i<up; i++){
                    cout<<res[i]<<" ";
                }
                cout<<endl;
            }
            
            sign1 = 0;
            sign2 = 0;
            data1.clear();
            data2.clear();
        }
        
    } 
    
}