CodeForces 156 D.Clues（Purfer序列）

Description

给出一张$n$个点$m$条边的图，要求加最少的边使得其连通，问方案数，结果模$k$

Input

第一行三个整数$n,m,k$，之后$m$行每行两个整数$u,v$表示一条边$(1\le n\le 10^5,0\le m\le 10^5,1\le k\le 10^9)$

Output

输出加最少边使得图连通的方案数，结果模$k$

Sample Input

2 0 1000000000

Sample Output

1

Solution

假设图有$cnt$个连通块，那么最少加$cnt-1$条边即可使得图连通，即把这些连通块连成一棵树，假设第$i$个连通块的点数为$num_i$，由$purfer$序列与树的一一对应关系可知$n$个节点的带标号树共有$n^{n-2}$棵，考虑两个连通块之间连边可以从其中的点随便取，故答案为$cnt^{cnt-2}\cdot\prod\limits_{i=1}^{cnt}num_i$，特判$cnt=1$的情况即可

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
int n,m,k,fa[maxn],num[maxn];
int find(int x)
{
    if(fa[x]==x)return x;
    return fa[x]=find(fa[x]);
} 
void unite(int x,int y)
{
    x=find(x),y=find(y);
    if(x==y)return ;
    fa[x]=y;
    num[y]+=num[x]; 
}
int Pow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1)ans=(ll)ans*a%k;
        a=(ll)a*a%k;
        b>>=1;
    }
    return ans;
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        for(int i=1;i<=n;i++)fa[i]=i,num[i]=1;
        while(m--)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            unite(u,v);
        }
        int ans=1,cnt=0;
        for(int i=1;i<=n;i++)
            if(fa[i]==i)cnt++,ans=(ll)ans*num[i]%k;
        if(cnt==1)ans=1;
        else ans=(ll)ans*Pow(n,cnt-2)%k;
        printf("%d\n",ans%k);
    }
    return 0;
}