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

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

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

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

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

#include
using namespace std;
typedef long long ll;
typedef pair 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<=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

 

你可能感兴趣的:(完全图最小生成树)