LeetCode 中级 - 优势洗牌(870)

时间:2023-03-10 06:56:35
LeetCode 中级 - 优势洗牌(870)

给定两个大小相等的数组 A 和 B,A 相对于 B 的优势可以用满足 A[i] > B[i] 的索引 i 的数目来描述。

返回 A 的任意排列,使其相对于 B 的优势最大化。

示例 2:

输入:A = [12,24,8,32], B = [13,25,32,11]
输出:[24,32,8,12] 思路: 类似田忌赛马,先将A排序。
class Solution {
public:
vector<int> advantageCount(vector<int>& A, vector<int>& B) {
vector<int> c;//存入符合要求的序列
sort(A.begin(),A.end());//先将A按递增排序
bool flag;
int index=;
while(!A.empty()&&index<B.size())
{
flag=false;
for(int i=;i<A.size();i++)
{
//找到第一个大于B[index]的位置
if(A[i]>B[index])
{
flag = true;
index++;
c.push_back(A[i]);//将其放入c中
A.erase(A.begin()+i);//移除该位置的值
break;//跳出循环
}
}
if(!flag)//若都比B[index]小,将A中最小的压入c
{
c.push_back(A[]);
A.erase(A.begin());
index++;
}
}
return c;
}
};