LeetCode 870. Advantage Shuffle

题解

这题有点田忌赛马的意思。对每个B中的数都找A中最小的大于它的数即可。

---

Code

这个map用得花

vector<int> advantageCount(vector<int> A, vector<int> B) {
        map<int, int> m;
        for (int i : A) m[i]++;
        map<int,int>::iterator it;
        vector<int> res;
        for (int i : B) {
            it = m.upper_bound(i);
            int x = it != m.end() ? it->first : m.begin()->first;
            if (--m[x] == 0) m.erase(x);
            res.push_back(x);
        }
        return res;
    }

糟糕的排序比较法

 vector<int> advantageCount(vector<int>& A, vector<int>& B) {
        vector<int> ac(A);
        vector<int> bc(B);
        sort(ac.begin(),ac.end());
        sort(bc.begin(),bc.end());
        int n = A.size();
        
        int count,k=0;
        while(k!=n-1){
            count=0;
            for(int i=n-1;i>=k;i--){
                if(ac[i]>bc[i-k]) count++;
            }
            if(count == n-k || count== n-k-1){
                break;
            }
            k++;
        }// k 偏移量
        
        vector<bool> vis(n,false);// 有重复
        int p;
        for(int i=0;i<n;i++){
            for(p=0;p<n;p++){
                if(bc[p]==B[i]&& !vis[p]){
                    vis[p]=true;
                    break;
                }
            }   
            A[i]=ac[(p+k+n)%n];// 映射
        }        
        return A;
    }