51nod1601 完全图的最小生成树计数

http://www.51nod.com/Challenge/Problem.html#problemId=1601

分治，最多只有30层，所以是30*n*30

对于字典树中的每一个点，我们是需要把他们的连通块连起来的，那么就找到两个连通块间的最小值，用字典树找，再统计得到最小值的方案数。

如果扫到底有一堆一样的，那么他们就可以随便连乘生成树，完全图生成树数量是n^(n-2)，所以直接乘到方案数里就行了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> p;
const int maxl=1e5+10;
const int mod=1e9+7;
const ll inf=1ll<<60;

ll n,tot,cnt,ans,ans1;
ll jc[maxl],num[maxl];
int a[maxl],s[maxl],t[maxl];
ll sum[maxl*30];
ll tr[maxl*30][2];

inline ll qp(ll a,ll b)
{
	ll ans=1,cnt=a;
	while(b)
	{
		if(b&1)
			ans=ans*cnt%mod;
		cnt=cnt*cnt%mod;
		b>>=1;
	}
	return ans;
}

inline void prework()
{
	scanf("%lld",&n);
	for(int i=2;i<=n;i++)
		num[i]=qp(i,i-2);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);		
}

inline void init()
{
	for(int i=0;i<=tot;i++)
		tr[i][0]=tr[i][1]=0,sum[i]=0;
	tot=0;
}

inline void insert(int x)
{
	int u=0,c;
	for(int i=30;i>=0;i--)
	{
		c=(x>>i)&1;
		if(!tr[u][c])
			tr[u][c]=++tot;
		u=tr[u][c];
	}
	sum[u]++;
}

inline p find(int x)
{
	int u=0,ans=0,c;
	for(int i=30;i>=0;i--)
	{
		c=(x>>i)&1;
		if(tr[u][c])
			u=tr[u][c];
		else
			u=tr[u][c^1],ans|=(1<<i);
	}
	return {ans,sum[u]};
}

inline void solv(int l,int r,int dep)
{
	if(l>=r) return;
	if(dep<0)
	{
		if(r-l+1>=2)
			ans1=ans1*num[r-l+1]%mod;
		return;
	}
	int cnt1=0,cnt2=0;
	for(int i=l;i<=r;i++)
	if((a[i]>>dep)&1)
		s[++cnt1]=a[i];
	else
		t[++cnt2]=a[i];
	for(int i=1;i<=cnt1;i++)
		a[l+i-1]=s[i];
	for(int i=1;i<=cnt2;i++)
		a[l+cnt1+i-1]=t[i];
	init();p tmp;ll mi=inf,cnt=0;
	for(int i=1;i<=cnt2;i++)
		insert(t[i]);
	for(int i=1;i<=cnt1;i++)
	{
		tmp=find(s[i]);
		if(tmp.first<mi)
			mi=tmp.first,cnt=tmp.second;
		else if(tmp.first==mi)
			cnt=(cnt+tmp.second)%mod;
	}
	if(mi!=inf && cnt) 
		ans+=mi,ans1=ans1*cnt%mod;
	solv(l,l+cnt1-1,dep-1);solv(l+cnt1,r,dep-1);
}

inline void mainwork()
{
	ans=0;ans1=1;
	solv(1,n,30);
}

inline void print()
{
	printf("%lld\n%lld\n",ans,ans1);
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}