hdu4334

hdu4334

题意：

给你5个数组，每个数组里有相同数量的元素，让你判断是否能在每个数组中选取1个元素使5个数相加为0

思路

把5个数组合并成3个（1,2,2），再进行排序贪心求解

事实上考虑如下问题，快速求解序列A，序列B中，是否有Ai+Bj=x ,记得是某知名公司的一道面试题吧，是两个指针的应用，将A，B升序排列，初试 i 指针指向A[1] ，j 指针指向 B[b_size] ,比较Ai + Bj 与 x 的大小。 若A[i]+B[j] < x , i 指针右移 ; 若 A[i]+B[j]>x , j 指针左移 。事实上是一个贪心的思想。(转自博客)

#include <queue>
#include <cstdlib>
#include <cmath>
#include <cstdio>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
using namespace std;

int main()
{
    int t,n;
    LL a[6][201],b[40001],c[40001];
    scanf("%d",&t);
    next:while(t--)
    {
    	scanf("%d",&n);
    	for(int i=1;i<=5;i++)
    	for(int j=1;j<=n;j++)
    	scanf("%lld",&a[i][j]);
    	int p=0;
    	for(int i=1;i<=n;i++)//合并集合 
    	for(int j=1;j<=n;j++)
    	b[++p]=a[1][i]+a[2][j];
    	int q=0;
    	for(int i=1;i<=n;i++)//同上 
    	for(int j=1;j<=n;j++)
    	c[++q]=a[3][i]+a[4][j];
    	sort(b+1,b+1+p);
    	sort(c+1,c+1+q);
    	for(int i=1;i<=n;i++)
    	{
    		a[5][i]=-a[5][i];
    		int l=1;//设两个指针 
    		int r=q;
    		while(l<=p&&r>=1)
    		{
    			if(a[5][i]==b[l]+c[r]){printf("Yes\n");goto next;}
    			else if(b[l]+c[r]<a[5][i])l++;
    			else r--;
			}
		}
    	
		printf("No\n");
	}
    return 0;
}