http://www.lydsy.com/JudgeOnline/problem.php?id=1016
我们来分析一下这个最小生成树
如果两颗最小生成树中至少有一条边不同,则这两个最小生成树就是不同的
因为说相同边权的边的编号不影响最小生成树的大小,这个我们就可以算进多个答案了
所以我们可以先把边排序,跑一遍Kruskal记录一下相同边权的边对答案产生影响的数目还有起始终止位置(排序后)
然后我们每种边权跑一个dfs,O(n)跑一遍暴力计算全部边能影响答案的数目
最后乘起来就好了
需要注意到的是,并查集不能状态压缩,否则dfs回溯时会产生错误
#include<bits/stdc++.h>
using namespace std;
const int MOD=31011;
struct edge{
int x,y,v;
}a[1001];
inline bool cmp(edge a,edge b){return a.v<b.v;}
int n,m,fa[101],sum=0,cnt=0,l[1001]={0},r[1001]={0},v[1001]={0};
inline int getfather(int v){
if(fa[v]==v)return v;
return getfather(fa[v]);
}
inline void dfs(int i,int x,int s){
if(x==r[i]+1){
if(s==v[i])cnt++;
return;
}
int xx=getfather(a[x].x),yy=getfather(a[x].y);
if(xx!=yy){fa[xx]=yy;dfs(i,x+1,s+1);fa[xx]=xx;}
dfs(i,x+1,s);
}
int ans=1;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].v);
sort(a+1,a+m+1,cmp);
for(int i=1;i<=n;i++)fa[i]=i;
for(int i=1;i<=m;i++){
if(a[i].v!=a[i-1].v)r[sum]=i-1,l[++sum]=i;
int xx=getfather(a[i].x),yy=getfather(a[i].y);
if(xx!=yy)fa[xx]=yy,cnt++,v[sum]++;
}
r[sum]=m;
if(cnt<n-1){
printf("0");
return 0;
}
for(int i=1;i<=n;i++)fa[i]=i;
for(int i=1;i<=sum;i++){
cnt=0;
dfs(i,l[i],0);
(ans*=cnt)%=MOD;
for(int j=l[i];j<=r[i];j++){
int xx=getfather(a[j].x),yy=getfather(a[j].y);
if(xx!=yy)fa[xx]=yy;
}
}
printf("%d",ans);
return 0;
}