HDU-5536 Chip Factory

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/C_13579/article/details/82666306

地址：http://acm.hdu.edu.cn/showproblem.php?pid=5536

思路：题目限时9s,且题目输入数据的范围也提示的很明显，就是枚举 x=a[i]+a[j],建立a[]的01字典树，再去找x的最大异或和，对于i!=j!=z,因此还要排除掉 a[i],a[j]的影响，对于树的搜索中可判断a[i],a[j]是否还在当前路径上，在就需要sum[u]>在的个数s，否则就排除掉不在的即可

Code :
 

#include<iostream>
using namespace std;

const int MAX_N=1005;
const int MAX_S=32005;
int n,T;
int a[MAX_N];
int Trie[MAX_S][2],num;
int Sum[MAX_S];

void Insert(int x);
int Query(int x1,int x2);
int main()
{
	scanf("%d",&T);
	for(int t=1;t<=T;++t)
	{
		for(int i=0;i<=num;++i)
			Trie[i][0]=Trie[i][1]=Sum[i]=0;
		num=0;
		scanf("%d",&n);
		int x;
		for(int i=0;i<n;++i)
		{
			scanf("%d",&a[i]);
			Insert(a[i]);
		}
		int ans=0;
		for(int i=0;i<n;++i)
			for(int j=0;j<i;++j)
				ans=max(ans,Query(a[i],a[j]));
		printf("%d\n",ans);
	}
	
	return 0;
}

void Insert(int x)
{
	int u=0,t;
	for(int i=31;i>=0;--i)
	{
		t=(x>>i)&1;
		if(!Trie[u][t])	Trie[u][t]=++num;
		u=Trie[u][t];
		++Sum[u];
	}
}

int Query(int x1,int x2)
{
	int x=x1+x2,ss=0,t1,t2;
	int u=0,t,p,res=0;
	bool bo1=true,bo2=true;
	for(int i=31;i>=0;--i)
	{
		t=(x>>i)&1;	ss=0;
		t1=(x1>>i)&1;	t2=(x2>>i)&1;
		if(bo1&&t1==!t)	++ss;
		if(bo2&&t2==!t)	++ss;
		p=Trie[u][!t];
		if(p&&Sum[p]>ss){
			res=res|(1<<i);
			if(bo1&&t1==t)	bo1=false;
			if(bo2&&t2==t)	bo2=false;
			u=p;
		}else{
			if(bo1&&t1==!t)	bo1=false;
			if(bo2&&t2==!t)	bo2=false;
			u=Trie[u][t];
		}
	}
	return res;
}