LeetCode 870.优势洗牌（C++）

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

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

 

示例 1：

输入：A = [2,7,11,15], B = [1,10,4,11]
输出：[2,11,7,15]

示例 2：

输入：A = [12,24,8,32], B = [13,25,32,11]
输出：[24,32,8,12]

 

提示：

1. 1 <= A.length = B.length <= 10000
2. 0 <= A[i] <= 10^9
3. 0 <= B[i] <= 10^9

 

思路：A排序之后，给B每个元素分配最小大于的元素，再删除A中该元素。若A中元素都比B中元素小，则分配A[0]

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> advantageCount(vector<int>& A, vector<int>& B) {
    vector<int> temp;
    sort(A.begin(), A.end());
    for (size_t i = 0; i < B.size(); ++i) {
        size_t j = 0;
        bool flag = false;
        while (j < A.size()) {
            if (A[j] > B[i]) {
                temp.push_back(A[j]);
                A.erase(A.begin() + j);
                flag = true;
                break;
            }
            ++j;
        }
        if (!flag) {
            temp.push_back(A[0]);
            A.erase(A.begin());
        }
    }
    return temp;
}

int main()
{
    vector<int> A = { 12,24,8,32 }, B = { 13,25,32,11 }, temp;
    temp = advantageCount(A, B);
    for (auto& i : temp) {
        cout << i << " ";
    }

    system("PAUSE");
    return 0;
}