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=0;
        while(!A.empty()&&index<B.size())
        {
            flag=false;
           for(int i=0;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[0]);
                A.erase(A.begin());
                index++;
            }
        }
        return c;
    }
};