思路:很简单,田忌赛马。强对强,弱对弱,如果发现你的强打不过对方的强,就用最弱的对对方的强;
程序实现问题:
1、为了分出强弱,肯定要sort()排序一下,
sort(C.rbegin(), C.rend()); 降序
sort(C.begin(), C.end()); 升序
2、排序后索引都变了,怎么存储索引,
定义
vector<pair<int,int>> C;
C.push_back({B[i], i});
pair对存值和索引,注意初始化,这不是双vector
3、交换vector中的次序,涉及到
A.insert(A.begin() + i, A[size - 1]); 插入在第i个元素之前
A.pop_back();
代码:
class Solution {
public:
vector<int> advantageCount(vector<int>& A, vector<int>& B) {
vector<pair<int,int>> C;
int size = A.size();
vector<int> result(size, -1);
for(int i = 0; i < size; i++)
{
C.push_back({B[i], i});
}
sort(C.rbegin(), C.rend());
sort(A.rbegin(), A.rend());
for(int i = 0; i < size; i++)
{
if(A[i] <= C[i].first)
{
A.insert(A.begin() + i, A[size - 1]);
A.pop_back();
}
}
for(int j =0; j< size; j++)
{
result[C[j].second] = A[j];
}
return result;
}
};