美团20230826秋招-小美的数组重排

题目描述与示例

题目描述

小美有两个长度为n的数组a和b。

小美想知道，能不能通过重排a数组使得对于任意1 <= i <= n, 1 <= ai+bi <= m？

将会有q次询问。

输入描述

第一行一个整数q (1 <= q <= 30)。表示询问次数。

对于每一个询问：

第一行输入两个整数n, m (1 <= n, m <= 500)

第二行输入n个整数ai (-500 <= ai <= 500)

第三行输入n个整数bi (-500 <= bi <= 500)

输出描述：

q行，每行输出一个字符串，如果能通过重排满足条件则输出"Yes"(不含引号)，否则输出"No"。

示例

输入

2
5 3
-1 -2 3 4 5
-1 3 4 2 5
5 6
-1 -2 3 4 5
-1 3 4 2 5

输出

No
Yes

说明

对于第一个用例，无论怎么重排都不满足条件。

对于第二个用例，将数组a重排为[5,3,-2,4,-1]时满足条件。

解题思路：

贪心的方法，首先将数组a进行升序排列，然后将数组b进行降序排列，再然后将数组a和b中同一位置的元素相加看是否满足1 <= ai+bi <= m。如果有一个不满足，则表示无论怎么重新排列都不满足条件，输出No，反之输出Yes。

这是为什么呢，因为a是升序排列，最小的在前，优先考虑数组a中小的元素。b是降序排列，最大的在前，优先考虑数组b中大的元素。

如果ai+bi > m，比如a0+b0 > m，那么就是说明b数组中的最大值太大了，连a中最小的和它相加都不满足条件，那么a中其他的元素肯定也不满足。

如果ai+bi < 1， 比如a0+b0 < 1，那么就说明a数组中的最小值太小了，连b中的最大值和它相加都不满足条件，那么b中的其他元素肯定也不满足。

那么如果一旦在某个位置i发生上述两种情况，就说明要返回No了。如果所有位置的元素都满足1 <= ai+bi <= m，则返回Yes。

C++代码

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

using namespace std;

bool solve(int n, int m, vector<int>& a, vector<int>& b) {
    sort(a.begin(), a.end());
    sort(b.rbegin(), b.rend());

    for (int i = 0; i < n; ++i) {
        if (!(1 <= a[i] + b[i] && a[i] + b[i] <= m)) {
            return false;
        }
    }

    return true;
}

int main() {
    int q;
    cin >> q;

    for (int i = 0; i < q; ++i) {
        int n, m;
        cin >> n >> m;

        vector<int> a(n);
        vector<int> b(n);

        for (int j = 0; j < n; ++j) {
            cin >> a[j];
        }

        for (int j = 0; j < n; ++j) {
            cin >> b[j];
        }

        bool result = solve(n, m, a, b);
        cout << (result ? "Yes" : "No") << endl;
    }

    return 0;
}

参考链接：大厂秋招真题【贪心】美团20230826秋招T2-小美的数组重排