题目描述

想了半天不知道py咋实现这个指针遍历顺序…for转while好麻烦
flag这种或st[] = true/false 以后要多练练

update:
貌似这个会快很多

vector<int> advantageCount(vector<int>& A, vector<int>& B) {      
        vector<pair<int,int>> v;
        for(int i=0;i<B.size();i++)
            v.push_back(make_pair(B[i],i));
        sort(v.begin(),v.end());
        sort(A.begin(),A.end());
        vector<int> arr;

        int i=0;
        int j=0;
        while(i<B.size()){
            if(A[i]>v[j].first){
                B[v[j].second]=A[i];
                j++;
                i++;
            }
            else{
                A[i]=-1*(A[i]+1);
                i++;
        }}

        for(int i=0;i<B.size();i++){
            if(A[i]<0){
                B[v[j].second]=-1*A[i]-1;
                j++;
                }
        }
        return B;
    }

或者

vector<int> advantageCount(vector<int>& A, vector<int>& B) {
        int n = A.size();
        if(n < 2 ) return A;

        sort(A.begin(), A.end());
        vector<pair<int,int>> C;
        for(int i=0; i<n; ++i)
            C.push_back( make_pair(B[i], i));    
        sort(C.begin(), C.end());

        int l = 0, r = n-1;
        vector<int> D(n); 
        for(int i = n-1; i >=0; i--){
           if(A[r]<=C[i].first)
                D[C[i].second] = A[l++];
            else
                D[C[i].second] = A[r--];       
        }   
        return D;
    }

旧版：每次在A里找B的最小值，如果找不到就返回A比它大的最小值。

vector<int> advantageCount(vector<int>& A, vector<int>& B) {
        sort(A.begin(), A.end());
        vector<int> res;
        //res.resize(A.size());
        for (int i = 0; i < B.size(); i++)
        {
            bool flag = false;
            for (int j = 0; j < A.size(); j++)
            {
                if (A[j] > B[i])
                {
                    res.push_back(A[j]);
                    A.erase(A.begin() + j);
                    flag = true;
                    break;
                }
            }
            if (!flag)
            {
                res.push_back(A[0]);
                A.erase(A.begin());
            }
        }
        return res;
    }