【51nod1806】wangyurzee的树(prufer序列)(容斥原理)

传送门


题解:

关于度数的限制我们可以考虑放到prufer序列上。

一个点的度数就是它在prufer序列中出现次数+1。

所以问题就变成了一个长度为 n − 2 n-2 n2的序列,每个位置的数可以是 [ 1 , n ] [1,n] [1,n]中间的任意整数,限制某些数出现次数不能为某个值,求方案数。

很显然的容斥,考虑用总方案数减去所有不合法的方案,就是求满足 0 , 1 , 2 ⋯ m 0,1,2\cdots m 0,1,2m个限制的所有方案数。

假设我们现在满足 ∣ S ∣ |S| S中的限制,设 x = ∣ S ∣ , s u m = ∑ i = 1 x S i x=|S|,sum=\sum\limits_{i=1}^xS_i x=Ssum=i=1xSi,考虑从 n − 2 n-2 n2个里面选出 s u m sum sum个来 ( n − 2 s u m ) {n-2\choose sum} (sumn2),然后这 s u m sum sum个位置要对 S S S做可重集合的排列 s u m ! ∏ S i ! \frac{sum!}{\prod S_i!} Si!sum!,剩下 n − 2 − s u m n-2-sum n2sum可以随便填剩下 n − x n-x nx个数 ( n − x ) n − 2 − s u m (n-x)^{n-2-sum} (nx)n2sum

所以方案数为: ( n − 2 ) ! ( n − 2 − s u m ) ! ∏ S i ! ( n − x ) n − 2 − s u m \frac{(n-2)!}{(n-2-sum)!\prod S_i!}(n-x)^{n-2-sum} (n2sum)!Si!(n2)!(nx)n2sum

然后容斥dfs就行了,注意可能有对于同一个点的限制,要在dfs的时候注意处理一下。

复杂度: O ( 2 m ) O(2^m) O(2m)


代码:

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;
using pii=std::pair<int,int>;
#define fi first
#define se second

cs int mod=1e9+7;
inline int add(int a,int b){return (a+=b)>=mod?a-mod:a;}
inline int dec(int a,int b){return (a-=b)<0?a+mod:a;}
inline int mul(int a,int b){ll r=(ll)a*b;;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline int Inc(int &a,int b){return (a+=b)>=mod&&(a-=mod),a;}
inline int Dec(int &a,int b){return (a-=b)<0&&(a+=mod),a;}

cs int P=1e6+6;
int fac[P],ifac[P];

int n,m,cnt;
int ans=0;

inline void init(){
	fac[0]=ifac[0]=1;
	for(int re i=1;i<=n;++i)fac[i]=mul(fac[i-1],i);
	ifac[n]=power(fac[n],mod-2);
	for(int re i=n-1;i;--i)ifac[i]=mul(ifac[i+1],i+1);
}

std::vector<int> d[20];
pii limit[20];

void dfs(int cur,int r1,int r2,int down,int coef){
	if(r1<0)return ;
	if(cur>cnt){
		Inc(ans,mul(coef,mul(mul(fac[n-2],down),power(r2,r1,ifac[r1]))));
		return ;
	}
	dfs(cur+1,r1,r2,down,coef);
	for(int re v:d[cur])dfs(cur+1,r1-v,r2-1,mul(down,ifac[v]),mod-coef);
}

signed main(){
//	freopen("tree.in","r",stdin);
	scanf("%d%d",&n,&m);init();if(n==1)cout<<1,exit(0);
	for(int re i=1;i<=m;++i)scanf("%d%d",&limit[i].fi,&limit[i].se);
	std::sort(limit+1,limit+m+1);
	for(int re i=1;i<=m;++i){
		if(limit[i].fi!=limit[i-1].fi)++cnt;
		d[cnt].push_back(limit[i].se-1);
	}
	dfs(1,n-2,n,1,1);
	cout<<ans<<"\n";
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值