bzoj1016 [JSOI2008]最小生成树计数(矩阵树定理,最小生成树)

Description
  现在给出了一个简单无向加权图。你不满足于求出这个图的最小生成树,而希望知道这个图中有多少个不同的
最小生成树。(如果两颗最小生成树中至少有一条边不同,则这两个最小生成树就是不同的)。由于不同的最小生
成树可能很多,所以你只需要输出方案数对31011的模就可以了。
Input
  第一行包含两个数,n和m,其中1<=n<=100; 1<=m<=1000; 表示该无向图的节点数和边数。每个节点用1~n的整
数编号。接下来的m行,每行包含两个整数:a, b, c,表示节点a, b之间的边的权值为c,其中1<=c<=1,000,000,0
00。数据保证不会出现自回边和重边。注意:具有相同权值的边不会超过10条。
Output
  输出不同的最小生成树有多少个。你只需要输出数量对31011的模就可以了。
Sample Input
4 6
1 2 1
1 3 1
1 4 1
2 3 2
2 4 1
3 4 1
Sample Output
8
HINT
Source


考虑 k r u s k a l kruskal kruskal算法
对于相同权值的边,如果没有加入意味着他连接的联通块已经联通了,那么对于这些边,顺序对联通性不影响,也就是选边的方案对后面没有影响

那么我们只要对每一种权值做一遍就行了,最后乘起来就是答案

复杂度上界 O ( n 4 ) O(n^4) O(n4)
(其实我代码是 O ( n 4 l o g n ) O(n^4logn) O(n4logn)的?

#include<bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = j;i <= k;++i)
#define repp(i,j,k) for(int i = j;i >= k;--i)
#define rept(i,x) for(int i = linkk[x];i;i = e[i].n)
#define P pair<int,int>
#define Pil pair<int,ll>
#define Pli pair<ll,int>
#define Pll pair<ll,ll>
#define pb push_back 
#define pc putchar
#define mp make_pair
#define file(k) memset(k,0,sizeof(k))
#define ll long long
int rd()
{
    int sum = 0;char c = getchar();bool flag = true;
    while(c < '0' || c > '9') {if(c == '-') flag = false;c = getchar();}
    while(c >= '0' && c <= '9') sum = sum * 10 + c - 48,c = getchar();
    if(flag) return sum;
    else return -sum;
}
const int p = 31011;
int n,m;
int f[110][110];
int F[110];
bool vis[110];
struct line{int l,r,v;}e[1010];
inline int del(int a,int b){return (a-b+p)%p;}
inline int mul(int a,int b){return 1ll*a*b%p;}
inline void mo(int &a){a = (a%p+p)%p;}
int getdet(int n)
{
	int det = 1,fl = 0;
	rep(i,1,n) rep(j,i+1,n)
	{	
		int A = f[i][i],B = f[j][i];
		while(B)
		{
			int t = A/B;A %= B;swap(A,B);
			rep(k,i,n) f[i][k] = del(f[i][k],mul(t,f[j][k]));
			rep(k,i,n) swap(f[i][k],f[j][k]);
			fl ^= 1;
		}
		if(f[i][i] == 0) return 0;
	}
	rep(i,1,n) det = mul(det,f[i][i]);
	return fl?(p-det):det;
}
bool mycmp(line x,line y){return x.v<y.v;}
int get(int x){return F[x] = F[x]==x?F[x]:get(F[x]);}
void merge(int x,int y){x = get(x);y = get(y);F[x] = y;}
void init()
{
	n = rd();m = rd();
	rep(i,1,m) e[i].l = rd(),e[i].r = rd(),e[i].v = rd();
	sort(e+1,e+m+1,mycmp);
    rep(i,1,n) F[i] = i;
}
int id[110];
int tmp[110],tot;
int main() 
{
	freopen("mst.in","r",stdin);
	freopen("mst.out","w",stdout);
	init();
	int i=1,j,ans=1;
	bool fl;
	while(i <= m)
	{
		j=i;
		while(j <= m) if(e[j].v != e[i].v){j--;break;} else j++;
		if(j > m) j--;
		fl = false;
		rep(k,i,j) if(get(e[k].l) != get(e[k].r)) fl = true;
		if(!fl) {i = j+1;continue;}
		file(f);file(vis);
		tot = 0;
		rep(k,1,n) if(!vis[get(k)]) vis[F[k]] = true,tmp[++tot] = F[k];
		sort(tmp+1,tmp+tot+1);
		rep(k,1,tot) id[tmp[k]] = k;
		rep(k,i,j)
		{
			int x = F[e[k].l],y = F[e[k].r];x = id[x];y = id[y];
			f[x][y]--;f[y][x]--;f[x][x]++;f[y][y]++;
		}
		rep(k,i,j) if(get(e[k].l) != get(e[k].r)) merge(e[k].l,e[k].r);
		int x = get(1);
		file(vis);
		rep(k,1,n) if(!vis[get(k)] && get(k) != x)
		{
			int t = get(k),y = id[x];vis[t] = true;t = id[t];
			f[t][y]--;f[y][t]--;f[t][t]++;f[y][y]++;
		}
		rep(k,1,tot) rep(t,1,tot) mo(f[k][t]);
		ans = 1ll*ans*getdet(tot-1)%p;
		i = j+1;
	}
	int id = get(1);
	rep(i,1,n) if(get(i) != id) {printf("0\n");return 0;}
	printf("%d\n",ans);
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值