2022牛客寒假算法基础集训营1 Jproblem

题意

牛牛是一个幼儿园老师，他经常带小朋友们一起做游戏。

现在，牛牛的班里有A个安静的小朋友和B个闹腾的小朋友，牛牛想要从中选出恰好nn个人来做游戏。这个游戏需要小朋友们手拉手围成一个圆圈，但不妙的是，如果两个闹腾的小朋友在圆圈中紧挨着，他们就会打闹，导致游戏无法进行。

每个小朋友还有一个幸福度v，若这位小朋友被选中参加游戏，则会使得班级的幸福度增加v。

请你求出，在满足上述所限制的情况下，恰当的安排围成圆圈的方法，能使得班级的幸福度最大为多少

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

输入描述:

输入第一行是一个整数T(1≤T≤10^3)，测试数据组数。

每组测试数据，第一行是三个整数A,B,n(2≤A,B≤10^4,3≤n≤A+B)，含义如题目所示。

第二行是A个数，第i个数vai​(1≤vai​≤104)表示某位安静小朋友的幸福度。

第三行是B个数，第i个数vbi​(1≤vbi​≤10^4)表示某位闹腾小朋友的幸福度。

此外，保证所有测试数据的(A+B)之和不会超过2∗10^5

输出描述:

每组测试用例，输出一行一个整数，表示最大幸福度。若无论如何安排都不能进行游戏，输出−1。

思路

见代码注释

#include<bits/stdc++.h>
using namespace std;
const int N=10010;
int a[N];
int b[N];
bool cmp(int a, int b) //用于sort函数，方便进行降序排列
{
    return a > b;
}
int main()
{
    int t,A,B,n;
    cin >> t;
    while (t--)
    {
        cin >> A >> B >> n;
        for (int i = 0; i < A; i++)cin >> a[i];
        for (int i = 0; i < B; i++)cin >> b[i];
        if (2 * A < n) //假设圆圈是由一个安静的小孩子和一个闹腾的小孩子连接在一起的那么至少圆圈一半是要由安静的小孩子构成，如果n是偶数的话，那就需要有（n+1）/2个安静的小朋友
        {
            cout << -1 << "\n"; //所以只要满足这个条件，就不能构成圆圈。
            continue;
        }
        sort(a, a + A, cmp);//降序排列
        sort(b, b + B, cmp);//降序排列
        int cnta = 0, cntb = 0; //cnta为 选择a的个数 cntb为选择b的个数
        long long sum = 0;
        while (cnta + cntb < n) //总选择数将会在最后一次循环到达n
        {
            if (cntb < cnta && cntb < B )//要保持选择的时候是a的数量大于b的数量的
            {
                if (a[cnta] >= b[cntb] && cnta < A)
                    sum += a[cnta++];
                else
                    sum += b[cntb++];
            }
            else//选不了b就只能选a了，挑着高的来选就行。
                sum += a[cnta++];
        }
        cout<<sum<<"\n";
    }
    return 0;
}