hdu4334

/*
分析:
    老题型了,合并集合就行了。
    第一次写的用了一个大FOR嵌一个小for,再在小for里面嵌一个
二分,TLE了,说明数据不水呀~。
    然后就是学来的方法了,看solve里面的p1和p2吧。


                                                           2012-12-13
*/











#include"stdio.h"
#include"string.h"
#include"stdlib.h"
#define N 201
int n;
__int64 num[4][N];
__int64 A[N*N],B[N*N],C[N];
int k1,k2;
int cmp(const void *a,const void *b)
{
	if(*(__int64 *)a>=*(__int64 *)b)	return 1;
	return -1;
}
void get_map()
{
	int i,l;
	int k;
	__int64 temp[N*N];
	scanf("%d",&n);

	for(i=0;i<4;i++)
	for(l=0;l<n;l++)
		scanf("%I64d",&num[i][l]);
	for(l=0;l<n;l++)	scanf("%I64d",&C[l]);

	k=0;
	for(i=0;i<n;i++)
	for(l=0;l<n;l++)
		temp[k++]=num[0][i]+num[1][l];
	qsort(temp,k,sizeof(__int64),cmp);
	A[0]=temp[0];
	for(k1=i=1;i<k;i++)
	{
		if(temp[i]==temp[i-1])	continue;
		A[k1++]=temp[i];
	}

	k=0;
	for(i=0;i<n;i++)
	for(l=0;l<n;l++)
		temp[k++]=num[2][i]+num[3][l];
	qsort(temp,k,sizeof(__int64),cmp);
	B[0]=temp[0];
	for(k2=i=1;i<k;i++)
	{
		if(temp[i]==temp[i-1])	continue;
		B[k2++]=temp[i];
	}
}
int solve()
{
	int i;
	int flag;
	int p1,p2;
	__int64 temp;
	flag=0;
	for(i=0;i<n;i++)
	{
		temp=-C[i];
		p1=0;p2=k2-1;
		while(p1<k1 && p2>=0)
		{
			if(A[p1]+B[p2]==temp)	{flag=1;break;}
			if(A[p1]+B[p2]>temp)	p2--;
			else					p1++;
		}
		if(flag)	break;
	}
	return flag;
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		get_map();
		if(solve())	printf("Yes\n");
		else		printf("No\n");
	}
	return 0;
}


你可能感兴趣的:(hdu4334)