codeforces 156D

传送门:https://codeforces.com/contest/156/problem/D

题意:给出一些联通块,加最少的边,使得这些连通块连通,问有多少种方案。

题解:裸的矩阵树定理。

#include
using namespace std;
typedef long long ll;
int fa[100005],num[100005];
int find(int t){
	return fa[t]==t?t:fa[t]=find(fa[t]);
}
int main(){
	int n,m,p;
	scanf("%d%d%d",&n,&m,&p);
	for(int i=1;i<=n;i++)num[i]=1,fa[i]=i;
	for(int i=1;i<=m;i++){
		int u,v;
		scanf("%d%d",&u,&v);
		int f1=find(u),f2=find(v);
		if(f1!=f2){
			fa[f1]=f2;
			num[f2]+=num[f1];
		}
	}
	ll ans=1;
	int k=0;
	for(int i=1;i<=n;i++){
		if(find(i)==i){
			ans=ans*num[i]%p;
			k++;
		}
	}
	for(int i=1;i<=k-2;i++){
		ans=ans*n%p;
	}
	printf("%lld\n",k==1?1ll%p:ans%p);
	return 0;
}

 

你可能感兴趣的:(并查集)