2020长春-Strange Memory-(启发式合并+位运算)

F

题意：
给你一个树，每个点有个权值<=1e6，然后问你有多少i和j满足va[i] xor va[j]==va[lca(i,j)]，然后对答案产生i xor j的答案，现在问你这个树的总贡献是多少。

思考：

1. 比赛的时候，然后就想反正肯定先固定一个点，然后去求，我先固定其中一个va[i]，那么和va[i]的lca，肯定就是va[i]上面所有的点。然后感觉这个题和之前牛客的推式子的那题牛客多校-Zero-chan差不多。都是关于树上某个点求好多个lca的问题。但是推了推发现不行，不一样，不能陷入那种思维。
2. 然后既然固定其中一个va[i]不好操作，那么我就固定lca，把这个点当做lca，那么只有以这个点为子树的下面的点才能组成lca为我。想到这里我就想到启发式合并了，就是子树问题。
3. 但是对于任意的两点，只有这两个点是now不同儿子子树中的点的时候，才能组成lca(i,j) = now。现在重儿子已经处理出来了，然后写add函数的时候，必须要一个儿子一个儿子的处理，因为这个点和全局变量中的点都必须是在不同子树中的，所以
   枚举所有轻儿子，先算答案，再把轻儿子加进去，写两个函数。
4. 对于算答案的时候怎么算呢？比如我这个点的权值为a，now的值为c，那么我想找权值为b = a xor c的点。如果用set[权值].insert(点编号)，这样每次找的时候还是n² 的复杂度(所有点权值都相同的时候)。然后我就想能不能直接用个数组存下来所有的编号和，但是异或没有结合律。
5. 但是之前看过关于位运算的题目，只要牵扯到位运算，几乎都是按位去处理。所以这里也可以把每个点按位存下来cnt[va[now]][i][0/1]就是权值为va[now],i位，为0，为1，的一共有多少个。那么求的时候，就直接先枚举自己的位情况和自己需要的权值的位情况累加到答案中就可以了。
6. 做这题的时候，混乱点就是对于以now为子树这个点，我肯定是先处理的重儿子，我就想重儿子的所有点和now这个点没有组成啊没算贡献。因为我想now的儿子和now的lca肯定是now，所以这个答案我想算上。实际上我要找va[spot] xor va[now] = va[now]的。所以va[spot] = 0，而实际上va[spot]=0是没有的，所以不用管now和别人的贡献。但是要把他给加到全局变量cnt中，因为先处理重儿子，然后处理轻儿子，这个now这个点没有处理，所以要加进去，虽然加进去对以now为lca的答案没有贡献，但是对以别的点为lca的是有贡献的。

代码：

#include<bits/stdc++.h>
#define ll long long
#define pb push_back
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);

using namespace std;

const int N = 1e5+10,M = 1e6+5;

int T,n,m;
int va[N];
int siz[N],son[N];
int cnt[M][23][2];
ll ans;

vector<int > e[N];

void get(int now,int p)
{
	siz[now] = 1;
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		get(spot,now);
		siz[now] += siz[spot];
		if(siz[spot]>siz[son[now]]) son[now] = spot;
	}
}

void count(int now,int p,int fa)
{
	int ned = va[now]^va[fa];
	if(ned<=1e6)
	{
		for(int i=0;i<=22;i++)
		{
			if(now>>i&1) ans += cnt[ned][i][0]*(1ll<<i);
			else ans += cnt[ned][i][1]*(1ll<<i);
		}
	}
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		count(spot,now,fa);
	}
}

void add(int now,int p,int value)
{
	for(int i=0;i<=22;i++) 
	{
		if(now>>i&1) cnt[va[now]][i][1] += value;
		else cnt[va[now]][i][0] += value;
	}
	for(auto spot:e[now])
	{
		if(spot==p) continue;
		add(spot,now,value);
	}
}

void dfs(int now,int p,int del)
{
	for(auto spot:e[now])
	{
		if(spot==p||spot==son[now]) continue;
		dfs(spot,now,1);
	}
	if(son[now]) dfs(son[now],now,0);
	for(int i=0;i<=22;i++)
	{
		if(now>>i&1) cnt[va[now]][i][1]++;
		else cnt[va[now]][i][0]++;
	}
	for(auto spot:e[now])
	{
		if(spot==p||spot==son[now]) continue;
		count(spot,now,now);
		add(spot,now,1);
	}
	if(del) add(now,p,-1);
}

signed main()
{
	IOS;
	cin>>n;
	for(int i=1;i<=n;i++) cin>>va[i];
	for(int i=1;i<n;i++)
	{
		int a,b;
		cin>>a>>b;
		e[a].pb(b);
		e[b].pb(a);
	}
	get(1,0);
	dfs(1,0,1);
	cout<<ans;
	return 0;
}

总结：
注意细节，一定要写清楚。