HDU 4334 Trouble （暴力）

Trouble

Time Limit: 5000MS

Memory Limit: 32768KB

64bit IO Format: %I64d & %I64u

Submit Status

Description

Hassan is in trouble. His mathematics teacher has given him a very difficult problem called 5-sum. Please help him. 
The 5-sum problem is defined as follows: Given 5 sets S_1,...,S_5 of n integer numbers each, is there a_1 in S_1,...,a_5 in S_5 such that a_1+...+a_5=0?

 

Input

First line of input contains a single integer N (1≤N≤50). N test-cases follow. First line of each test-case contains a single integer n (1<=n<=200). 5 lines follow each containing n integer numbers in range [-10^15, 1 0^15]. I-th line denotes set S_i for 1<=i<=5.

 

Output

For each test-case output "Yes" (without quotes) if there are a_1 in S_1,...,a_5 in S_5 such that a_1+...+a_5=0, otherwise output "No".

 

Sample Input

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

 

Sample Output

No Yes

 

Source

2012 Multi-University Training Contest 4

题意：给定五个集合，每个集合有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;
}