小朋友做游戏题解(前缀和)

 

题目描述

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

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

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

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

输入描述:

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

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

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

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

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

输出描述:

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

示例1

输入

3
3 6 7
1 3 4
5 4 3 4 3 5
4 6 7
1 3 4 1
5 4 3 4 3 5
7 7 7
1 2 3 4 5 6 7
9 8 7 6 5 4 3

输出

-1
23
46

思路:

1,先判断简单的,安静人数>=(n+1/2);

2,注意题干只要求闹腾小孩不相邻,数量不确定

3,前缀和,遍历判断啥时候最合适

代码:

#include<bits/stdc++.h>
using namespace std;
const int maxj=1e4+100;
typedef long long ll;
template<class t>inline void read(t &res){
    char c;t flag=1;
    while((c=getchar())<'0'||c>'9')if(c='-')flag=-1;res=c-'0';
    while((c=getchar())>='0'&&c<='9')res=res*10+c-'0';res*=flag;
}//快读
bool cmp(int a,int b){
    return a>b;
}
int x[maxj],y[maxj];
int main(){
    int t;
    read(t);
    while(t--){
        int a,b,n;
        read(a);read(b);read(n);
        for(int i=1;i<=a;++i){
                read(x[i]);
            }
        for(int i=1;i<=b;++i){
                read(y[i]);
            }
        if(a<(n+1)/2)
            cout<<-1<<'\n';
        else{
            int ans=0;
            sort(x+1,x+1+a,cmp);
            sort(y+1,y+1+b,cmp);
            for(int i=1;i<=a;++i){
                x[i]+=x[i-1];
            }
            for(int i=1;i<=b;++i){
                y[i]+=y[i-1];//前缀和
            }
            int m=min(n/2,b);//最大闹腾小孩数
            for(int i=0;i<=a;++i){
                int j=n-i;
                if(j<0||j>m)continue;
                ans=max(ans,x[i]+y[j]);//选最大
            }
            cout<<ans<<'\n';
        }
    }
    return 0;
}