HDU 4334 Trouble （暴力）

Posted 2020-07-03 Ritchie丶

题意：给定五个集合，每个集合有n个数，从每个集合各取一个数使和为0。如果能输出Yes，不能则输出No。

　　　　看了网上的题解有很多解法，个人感觉这题暴力就能过。

　　　　首先将一二两个集合合并为数组a，再将三四两个集合合并数组b，然后进行从小到大排序，

　　　　设第五个数组为数组c，依次遍历c中的每一个数，看在a，b中是否存在两个数的和与c中的数的和为0，

　　　　将数组a，b进行sort排序，然后一个正序即从小到大遍历，一个逆序从大到小遍历，

　　　　如果三个数的和小于0，a数组后移一位，否则b后移一位。

　　　　这题WA了两次，第一次WA后发现else if用的是三个i，忘了改成i，j，k了。

　　　　第二次WA后发现Yes输出成了YES，No输出成了NO。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
long long a[40005],b[40005],c[205],d[205],e[205],f[205],g[205];
long long i,j,x,k,l,n,t,flag=0,p;
int main()
{
    scanf("%lld",&t);
    while(t--)
    {
        flag=0;
        //输入
        scanf("%lld",&n);
        for(i=0;i<n;i++)
        scanf("%lld",&d[i]);
        for(i=0;i<n;i++)
        scanf("%lld",&e[i]);
        for(i=0;i<n;i++)
        scanf("%lld",&f[i]);
        for(i=0;i<n;i++)
        scanf("%lld",&g[i]);
        for(i=0;i<n;i++)
        scanf("%lld",&c[i]);
        //合并
        int cnt1=0;
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
            a[cnt1++]=d[i]+e[j];
        int cnt2=0;
        for(i=0;i<n;i++)
            for(j=0;j<n;j++)
            b[cnt2++]=f[i]+g[j];  //三四合并
        sort(a,a+cnt1);
        sort(b,b+cnt2);
        //计算
        for(i=0;i<n;i++) //第五个数组 c[]
        {
            j=0;
            k=cnt2-1;
             while(j<cnt1 && k>=0)
             {
                 if(a[j]+b[k]+c[i]==0LL)
                 {
                     flag=1;
                     break;
                 }
                 else if(a[j]+b[k]+c[i]<0)
                     j++;
                 else
                     k--;
             }

        }
        if(flag) cout<<"Yes"<<endl;
        else cout<<"No"<<endl;
    }
    return 0;
}