2020ICPC济南J Tree Constructer

题目链接

题目：

给定一棵$n$个结点的树，现要你将每个结点$u$赋上一个权值$a_u$，使得若$x,y$之间有边，则$a_{x}or{a}_y=2^{60}-1$，若$x,y$之间没有边，则$a_{x}or{a}_y\ne 2^{60}-1$，输出$a_{1...n}$。
$(1\le n\le n,0\le a_i<2^{60})$

题解：

一道挺妙的构造题。
问题分成两种情况，有边和无边，混在一起考虑比较困难，想到分开来考虑，可以想到二分图的模型。树本身就是一个二分图（没有奇环）。
将树上的点黑白染色后分成两个部分。先来考虑两个部分内部的情况，可以将白点的最末两位赋上01，将黑点的最末两位赋上10，这样白点内部和黑点内部就能满足限制了，且白点和黑点的最后两位或起来为11，而边只能是一端白点，一端黑点，所以也满足限制。再来考虑黑白之间的情况，可以从黑部和白部中点数较少的那个部分着手，不妨设黑部点数较少，我们将第一个黑点的高58位赋上111…1110，第二个赋上111…1101，第三个赋上111…1011，依此类推，因为黑部点数较少，所以少于等于50个，所以58位足以标识所有的点。然后对于每个白点，它连了几个黑点，就把那几个黑点高58位中为0的位置赋上1，其余位置赋上0，即可满足限制。

复杂度：$O(n)$

代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<string>
#include<bitset>
#include<sstream>
#include<ctime>
//#include<chrono>
//#include<random>
//#include<unordered_map>
using namespace std;

#define ll long long
#define ls o<<1
#define rs o<<1|1
#define pii pair<int,int>
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define sz(x) (int)(x).size()
#define all(x) (x).begin(),(x).end()
const double pi=acos(-1.0);
const double eps=1e-6;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const int maxn=105;
ll read(){
	ll x=0,f=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
int n;
vector<int>g[maxn],ve[2];
ll a[maxn];
void dfs(int u,int fa,int flag){
	ve[flag].pb(u);
	for(auto v:g[u]){
		if(v==fa)continue;
		dfs(v,u,flag^1);
	}
}
int test(){
	int flag=1;
	for(int i=1;i<=n;i++){
		for(int j=0;j<sz(g[i]);i++){
			int v=g[i][j];
			if((a[i]|a[v])!=((1LL<<60)-1)){
				flag=0;
				break;
			}
		}
		if(!flag)break;
	}
	return flag;
}
int main(void){
	// freopen("in.txt","r",stdin);
	scanf("%d",&n);
	int x,y;
	for(int i=1;i<=n-1;i++){
		scanf("%d%d",&x,&y);
		g[x].pb(y);
		g[y].pb(x);
	}
	dfs(1,0,1);
	for(int i=0;i<sz(ve[0]);i++){
		a[ve[0][i]]=1;
	}
	for(int i=0;i<sz(ve[1]);i++){
		a[ve[1][i]]=2;
	}
	if(sz(ve[0])>sz(ve[1])){
		swap(ve[0],ve[1]);
	}
	for(int i=0;i<sz(ve[0]);i++){
		a[ve[0][i]]+=((1LL<<58)-1)<<2;
		a[ve[0][i]]-=1LL<<(i+2);
		for(auto v:g[ve[0][i]]){
			a[v]+=1LL<<(i+2);
		}
	}
	for(int i=1;i<=n;i++)printf("%lld%c",a[i],(i==n)?'\n':' ');
	return 0;	
}