equal.txt

来自「以前ACM在哈工程HRBEU做的一些简单题目」· 文本 代码 · 共 51 行

#include <iostream.h>
#include <stdlib.h>
int cmp(const void *a,const void *b)
{
	return (*(long *)a)<(*(long *)b)?1:-1;
}
int main()
{
	long n,n1,a[30],i,sum,b,j,k,f,s,t;
	cin>>n;
	while(n--)
	{
		cin>>n1;
		i=0;
		sum=0;
		while(n1--)
		{
			cin>>a[i];
			sum+=a[i];
			i++;
		}
		f=0;
       if(sum%6==0)
	   {
		   qsort(a,i,sizeof(a[0]),cmp);
	       b=0;
	       s=0;
		   for(j=0,k=0;j<i;j++)
		   {
			   if(a[j]==-1)continue;
			   while(1)
		   {
			   if(a[j+k]==-1){k++;continue;}
			   if(j+k>=i)break;
			   s+=a[j+k];
			   t=a[j+k];
			   a[j+k]=-1;
			   if(s==sum/6){b++;s=0;k=0;break;}
			   else if(s>sum/6){a[j+k]=t;s=s-t;k++;}
			   else k++;
		   }
		   }
		  if(b==6)f=1;
	   }
	   if(f)cout<<"yes"<<endl;
	   else cout<<"no"<<endl;
	   }
	return 0;
}