1. 程式人生 > >LeetCode 870. Advantage Shuffle

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; }