[JSOI2008]最小生成树计数

题面
最小生成树计数题。
每个\(MST\)中,每种长度的边的数目是一定的,由此可以搞一下
因为限制了同一长度的边的数量不超过10,可以\(dfs\)

注意的一点:不要随随便便把东西放到for循环的里面如\(for(i=1;i<=cnt;i++,sum=0)\)qwq只有我这种弱鸡才犯的错

(洛谷数据真的水)(flag:之后用矩阵树定理搞一次这道题)

/**************************************************************
    Problem: 1016
    User: 568877201
    Language: C++
    Result: Accepted
    Time:12 ms
    Memory:860 kb
****************************************************************/
#include <cstdio>
#include <algorithm>
int n,m,ecnt,fa[105],ans,cnt,ncnt,sum;
const int mod=31011;
struct Edge{int x,y,val;}e[2005];
struct Count{int l,r,use;}g[1005];
void add(int bg,int ed,int v){e[++ecnt].x=bg;e[ecnt].y=ed;e[ecnt].val=v;}
bool cmp(Edge a,Edge b){return a.val<b.val;}
int find(int x){return x==fa[x]?x:find(fa[x]);}
void dfs(int x,int now,int k) {
	if(now==g[x].r+1){if(g[x].use==k)sum++;return;}
	int p=find(e[now].x),q=find(e[now].y);
	if(p!=q){
		fa[p]=q;
		dfs(x,now+1,k+1);
		fa[p]=p;fa[q]=q;
	}
	dfs(x,now+1,k);
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=0; i<105; i++) fa[i]=i;
	for(int i=1,a,b,c; i<=m; i++)
		scanf("%d%d%d",&a,&b,&c),add(a,b,c);
	std::sort(e+1,e+1+ecnt,cmp);	
	for(int i=1;i<=m;i++){
		if(e[i].val!=e[i-1].val){g[++cnt].l=i;g[cnt-1].r=i-1;}
		int p=find(e[i].x),q=find(e[i].y);
		if(p!=q) fa[p]=q,g[cnt].use++,ncnt++;
	}
	if(ncnt!=n-1){puts("0");return 0;}
	g[cnt].r=m;ans=1;
	for(int i=0; i<105; i++) fa[i]=i;
	for(int i=1;i<=cnt;i++){
		sum=0; 
		dfs(i,g[i].l,0);
		ans=(ans*sum)%mod;
		for(int j=g[i].l;j<=g[i].r;j++){
		int p=find(e[j].x),q=find(e[j].y);
		if(p!=q) fa[p]=q;
		}
	}
	printf("%d",ans);
	return 0;
}
posted @ 2018-07-01 20:10  SWHsz  阅读(120)  评论(0编辑  收藏  举报