hdu4334-Trouble-多校4-1004-题解

Brief Description:

给定5个集合，每个集合内元素个数<=200，先从每个集合中各取一个数，若能使这些数的和为0，输出Yes,否则输出No

Analysis:

把第1,2个集合的数和的所有结果按从小到大的顺序存在M[0]中，M[0]中元素总数为sum0

把第3,4个集合的数和的所有结果按从小到大的顺序存在M[1]中，M[1]中元素总数为sum1

从第5个集合中取出一个数key,查找M[0][i]+M[1][j]=key,

显然查找可以在线性时间内完成，令i=0，j=sum1-1,

若M[0][i]+M[1][j]>key,则j--，若M[0][i]+M[1][j]<key,则i++，一直循环直到找到M[0][i]+M[1][j]==key或i，j超出范围为止。

具体见以下代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <set>
#include <algorithm>

using namespace std;
typedef long long LL;
const int maxn = 205;
LL M[3][maxn*maxn];
LL a[5][maxn];
int n,sum0,sum1;
int main()
{
    int cas;
    scanf("%d",&cas);
    while(cas--){
         int i,j,k;
         scanf("%d",&n);
         for(i=0; i<5; i++){
            for(j=0; j<n; j++) scanf("%I64d",&a[i][j]);
         }
         set<LL>S1;   set<LL>S2;
         for(i=0; i<n; i++)
         for(j=0; j<n; j++){
            S1.insert(a[0][i]+a[1][j]);
            S2.insert(a[2][i]+a[3][j]);
         }
         set<LL>::iterator iter;
         sum0 = sum1 = 0;
         for(iter=S1.begin(); iter!=S1.end(); iter++){
            M[0][sum0++] = (*iter);
         }
         for(iter=S2.begin(); iter!=S2.end(); iter++){
            M[1][sum1++] = (*iter);
         }

         for(k=0; k<n; k++){
            LL key = -a[4][k];
            i = 0;    j = sum1-1;
            while(M[0][i]+M[1][j] != key){
                if(M[0][i] + M[1][j] > key)
                  j--;
                else if(M[0][i] + M[1][j] < key)
                       i++;
                if(j < 0 || i >= sum0) break;
            }
            if(M[0][i]+M[1][j] == key){
              puts("Yes"); break;
            }
         }
         if(k >= n) puts("No");
    }
    return 0;
}