LeetCode 870. Advantage Shuffle
阿新 • • 發佈:2018-12-16
題解
這題有點田忌賽馬的意思。對每個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;
}