J 小朋友做游戏——牛客训练营【前缀和C语言】

题目

 

 

思考与总结 

没错这又是一份菜鸟笔记！！！！与君共勉

我最开始思路是用sort函数将a数组（即安静的小孩）与b数组（淘气的小孩子）的幸福值按从大到小排序【不会还有人用冒泡排序吧，赶紧去学一下c++的sort函数】。

再用一个for循环一个一个来取a数组和b数组的较大值

但是唔，随着参与游戏的小朋友人数越来越多，for循环会超时

此时应该怎么办呢？【枯】

没错，用到前缀和优化时间复杂度【其实我也不太理解但是没关系以后会学】

由于两个数组已经按降序排列完成了，所以只需要枚举a数组（安静小孩）人数

那怎么判断无法玩游戏的-1情况呢？那就找找规律吧，当围成一个圈是安静的小孩和吵闹的小孩一人间隔一下时吵闹的小孩最多n/2人

#include<bits/stdc++.h>        //别看我不会c++，但是我喜欢c++万能开头来用c++的函数
using namespace std;
const int N=10010;
int main()
{
    int t;
    int a[N],b[N];
    scanf("%d",&t);
    while(t--)
    {
        memset(a,sizeof(a),0);
        memset(b,sizeof(b),0);                //string函数库的函数，数组每个元素重置为0
        int A,B,n;
        int suma[N]={0},sumb[N]={0};            //a，b数组前缀和
        scanf("%d %d %d",&A,&B,&n);

        for(int i=1;i<=A;i++)                //输入数组
            scanf("%d",&a[i]);
        for(int i=1;i<=B;i++)
            scanf("%d",&b[i]);

        if(A<n-n/2)                        //特判一下
            printf("-1\n");
        else
        {
            sort(a+1,a+A+1,greater<int>());
            sort(b+1,b+B+1,greater<int>());    //sort函数排序
            int i,j,x=n-n/2;
            int sum1,sum2;
            for(i=1;i<=A;i++)
                suma[i]=suma[i-1]+a[i];
            for(i=1;i<=B;i++)
                sumb[i]=sumb[i-1]+b[i];        //前缀和公式
            while(x<=n)
            {
                
                sum1=suma[x]+sumb[n-x];
                x++;
                sum2=suma[x]+sumb[n-x];
                if(sum1>sum2)                //sum会慢慢递减
                {
                    break;
                }
            }
            printf("%d\n",sum1);
        }
    }
    return 0;
}

菜鸟题解，如果有更妙方法，希望大佬可以指点一二，万分荣幸！！！