LOJ 6469 Magic trie树

题目链接

题意：
给你$n$个数，每一对数对答案的贡献是两个数的异或值在十进制下的位数，求总贡献。$n<=5e4$，数值范围是$1e18$

题解：
这个题主要是第一步转化不好想。另外还是先%%%y_immortal大佬，又是大佬给我推荐的题，然后我又是照着大佬的博客的题解学的。

第一步转化是考虑对于每一个$10^i$对答案的贡献，我们考虑有多少对数异或后比$10^i$大，那么就相当于有多少对数异或之后有$i+1$这一位，其实我们相当于在十进制下按位考虑对答案的贡献。

有了这步转化之后，如果你能往trie树方面想，应该思维难度就不是那么大了。我们先把$n$个数的01trie树建好，然后枚举$10^i$，考虑如何计算每一个$10^i$对答案的贡献。我们的计算方法是对于每一个数在01trie上遍历一遍，我们从高位到低位建01trie，所以也是从高位到低位建，并且在trie树的节点上维护一个子树size。如果$10^i$在某一位是0，那么如果与当前数异或后当前位会变成1的一定都是有贡献的，那么直接加上到1的子树size，然后尝试往异或后是0的方向走，看还有没有能有贡献的情况。如果$10^i$当前位是0也同理做一个简单的讨论就可以了。

这样的复杂度是$n\ast 18\ast 64$，大约是5e7，是可以通过的。

代码：

#include <bits/stdc++.h>
using namespace std;

int n,cnt;
long long a[50010],ans;
struct node
{
	int vis[2],sz;
}t[6000010];
inline long long read()
{
	long long x=0;
	char s=getchar();
	while(s>'9'||s<'0')
	s=getchar();
	while(s>='0'&&s<='9')
	{
		x=x*10+s-'0';
		s=getchar();
	}
	return x;
}
inline void build(long long x)
{
	int rt=1;
	for(int i=62;i>=0;--i)
	{
		if(x&(1ll<<i))
		{
			if(!t[rt].vis[1])
			t[rt].vis[1]=++cnt;
			rt=t[rt].vis[1];
		}
		else
		{
			if(!t[rt].vis[0])
			t[rt].vis[0]=++cnt;
			rt=t[rt].vis[0];
		}
		t[rt].sz++;
	}
}
inline long long ksm(long long x,long long y)
{
	long long res=1;
	while(y)
	{
		if(y&1)
		res=res*x;
		x=x*x;
		y>>=1;
	}
	return res;
}
inline long long query(long long x,long long y)
{
	int rt=1;
	long long res=0;
	for(int i=62;i>=0;--i)
	{
		if(!rt)
		break;
		if(x&(1ll<<i))
		{
			if(y&(1ll<<i))
			rt=t[rt].vis[0];
			else
			{
				res+=t[t[rt].vis[0]].sz;
				rt=t[rt].vis[1];
			}
		}
		else
		{
			if(y&(1ll<<i))
			rt=t[rt].vis[1];
			else
			{
				res+=t[t[rt].vis[1]].sz;
				rt=t[rt].vis[0];
			}
		}
	}
	return res;
}
int main()
{
	n=read();
	cnt=1;
	for(int i=1;i<=n;++i)
	{
		a[i]=read();
		build(a[i]);
	}
	for(int i=0;i<=18;++i)
	{
		long long ji=ksm(10,i)-1;
		for(int j=1;j<=n;++j)
		ans+=query(a[j],ji);
	}
	ans/=2;
	printf("%lld\n",ans);
	return 0;
}