2017 Multi-University Training Contest - Team 3：1004. Kanade's trio（01字典树）

感觉很多时候字典树都是这种题的万能解

（具体规则or什么是字典树：http://blog.csdn.net/jaihk662/article/details/53930927）

建立两棵01字典树，对于当前第j个节点，将前面所有数加入第一棵字典树，将后面所有数加入第二棵字典树

之后查询就好了，如果两个数a[i]和a[k]前x位都相同，那么很显然这两个数异或a[j]前x位也都相同

所以你只要找出a[i]和a[k]第一个不同的地方进行异或比较就好

在字典树上的操作就是：对于两棵字典树所有相同的前缀，答案加上两棵树下一个不同的节点的权值积

例如：两棵树都经过了10010(root->R->L->L->R->L)，那么

ans += sum1(100101)*sum2(100100)或者ans += sum1(100100)*sum2(100101)（具体加哪一个取决于a[j]）

但相同前缀个数可能过多，所以要在树上求后缀

tre[]：字典树

val[now][t]：第t棵字典树now节点的权值（t=0那棵用来存前面的数）

frd[x]：节点x的兄弟（x和frd[x]父亲是同一个）

rek[i][]：存后缀，前i-1位都相同第i个数不同对答案的贡献

#include<stdio.h>
#include<string.h>
#define LL long long
int root, tre[3000055][2], val[3000055][2], rek[35][2], a[560000];
LL frd[3000055];
void Insert(LL x, int t)
{
	int i, c, now = 0;
	for(i=30;i>=0;i--)
	{
		c = 0;
		if(x&(1<<i))
			c = 1;
		if(tre[now][c]==0)
		{
			tre[now][c] = ++root;
			if(tre[now][c^1])
			{
				frd[root] = tre[now][c^1];
				frd[tre[now][c^1]] = root;
			}
		}
		now = tre[now][c];
		val[now][t] += 1;
		rek[i][c^t] += val[frd[now]][t^1];
	}
}
void Delete(LL x, int t)
{
	int c, i, now = 0;
	for(i=30;i>=0;i--)
	{
		c = 0;
		if(x&(1<<i))
			c = 1;
		now = tre[now][c];
		val[now][t] -= 1;
		rek[i][c^t] -= val[frd[now]][t^1];
	}
}
LL Query(LL x)
{
	int i, c;
	LL sum = 0;
	for(i=30;i>=0;i--)
	{
		c = 0;
		if(x&(1<<i))
			c = 1;
		sum += rek[i][c];
	}
	return sum;
}
int main(void)
{
	LL ans;
	int T, n, i;
	scanf("%d", &T);
	while(T--)
	{
		ans = root = 0;
		memset(tre, 0, sizeof(tre));
		memset(val, 0, sizeof(val));
		memset(rek, 0, sizeof(rek));
		memset(frd, 0, sizeof(frd));
		scanf("%d", &n);
		for(i=1;i<=n;i++)
			scanf("%d", &a[i]);
		for(i=2;i<=n;i++)
			Insert(a[i], 1);
		Insert(a[1], 0);
		for(i=2;i<=n;i++)
		{
			Delete(a[i], 1);
			ans += Query(a[i]);
			Insert(a[i], 0);
		}
		printf("%lld\n", ans);
	}
}