bzoj1016 [JSOI2008]最小生成树计数——Kruskal+矩阵树定理

时间:2022-11-23 20:58:08

题目:https://www.lydsy.com/JudgeOnline/problem.php?id=1016

从 Kruskal 算法的过程来考虑产生多种方案的原因,就是边权相同的边有一样的功能,从而带来了多种选择;

对于每一层次(边权相同)的边来说,它们最终会把图进一步连通;

所以在这一层之前缩好点,看看这一层连接出几个新连通块,对于每个连通块内部做矩阵树定理求生成树个数,再乘法原理乘起来即可;

注意高斯消元的矩阵不能直接用原图的点标号等,求行列式会出错;

疑惑:以及高斯消元 return 时为什么要加个 abs?

代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;
vector<int>v[];
int n,m,fa[],pa[],a[][],c[][],ans=,mod=;
bool vis[];
struct N{
int hd,to,w;
N(int h=,int t=,int w=):hd(h),to(t),w(w) {}
}edge[];
bool cmp(N x,N y){return x.w<y.w;}
int find(int x,int f[]){return f[x]==x?x:find(f[x],f);}//
int gauss(int n)
{
for(int i=;i<=n;i++)
for(int j=;j<=n;j++)
a[i][j]%=mod;//
int fl=,ret=;
for(int i=;i<=n;i++)
{
int t=i;
for(int j=i+;j<=n;j++)
if(abs(a[j][i])>abs(a[t][i]))t=j;//abs
if(t!=i)
{
fl^=;
for(int j=i;j<=n;j++)swap(a[i][j],a[t][j]);
}
for(int j=i+;j<=n;j++)
while(a[j][i])
{
int tmp=a[i][i]/a[j][i];
for(int k=i;k<=n;k++)
{
int tp=a[i][k]; a[i][k]=a[j][k];//a=b
a[j][k]=(tp-a[j][k]*tmp)%mod;//b=a%b
}
fl^=;
}
(ret*=a[i][i])%=mod;
}
return (abs(ret)%mod+mod)%mod;//abs!?
// return ret;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=;i<=n;i++)fa[i]=i;
for(int i=,x,y,z;i<=m;i++)
scanf("%d%d%d",&edge[i].hd,&edge[i].to,&edge[i].w);
sort(edge+,edge+m+,cmp);
for(int i=;i<=m+;i++)
{
if(edge[i].w!=edge[i-].w || i==m+)
{
for(int j=;j<=n;j++)
{
if(!vis[j])continue;
int f1=find(j,pa);
v[f1].push_back(j);//v是点的集合
vis[j]=;
}
for(int j=;j<=n;j++)
{
if(v[j].size()<=)continue;
memset(a,,sizeof a);
int siz=v[j].size();
for(int k=;k<siz;k++)
for(int l=k+;l<siz;l++)
{
int x=v[j][k],y=v[j][l],t=c[x][y];
// a[x][x]+=t; a[y][y]+=t;
// a[x][y]-=t; a[y][x]-=t;
a[k+][k+]+=t; a[l+][l+]+=t;
a[k+][l+]-=t; a[l+][k+]-=t;//!
}
(ans*=gauss(siz-))%=mod;
// (ans*=gauss(n-1))%=mod;
for(int k=;k<siz;k++)fa[v[j][k]]=j;
}
for(int j=;j<=n;j++)
{
pa[j]=fa[j]=find(j,fa);
v[j].clear();
}
}
int f1=find(edge[i].hd,fa),f2=find(edge[i].to,fa);
if(f1==f2)continue;
pa[find(f1,pa)]=find(f2,pa); vis[f1]=; vis[f2]=;
c[f1][f2]++; c[f2][f1]++;
}
for(int i=;i<=n;i++)//!
if(find(i,fa)!=find(i-,fa)){printf(""); return ;}
printf("%d",ans);
return ;
}