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,000。数据保证不会出现自回边和重边。注意:具有相同权值的边不会超过10条。
Output
输出不同的最小生成树有多少个。你只需要输出数量对31011的模就可以了。
Sample Input
1 2 1
1 3 1
1 4 1
2 3 2
2 4 1
3 4 1
Sample Output
#include<cstdio> #include<iostream> #include<cstring> #include<cstdlib> #include<algorithm> #include<cmath> #include<queue> #include<deque> #include<set> #include<map> #include<ctime> #define LL long long #define inf 0x7ffffff #define pa pair<int,int> #define mod 31011 using namespace std; inline LL read() { LL x=0,f=1;char ch=getchar(); while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();} while(ch>=‘0‘&&ch<=‘9‘){x=x*10+ch-‘0‘;ch=getchar();} return x*f; } struct edge{int x,y,z;}e[100010]; inline bool operator <(const edge &a,const edge &b){return a.z<b.z;} struct seg{int l,r,v;}a[100010]; int n,m,cnt,tot,sum,ans=1; int fa[100010]; inline int getfa(int x){return fa[x]==x?x:getfa(fa[x]);} inline void dfs(int x,int now,int k) { if (now==a[x].r+1) { if (k==a[x].v)sum++; return; } int fx=getfa(e[now].x),fy=getfa(e[now].y); if (fx!=fy) { fa[fx]=fy; dfs(x,now+1,k+1); fa[fx]=fx;fa[fy]=fy; } dfs(x,now+1,k); } int main() { n=read();m=read(); for (int i=1;i<=m;i++) { e[i].x=read(); e[i].y=read(); e[i].z=read(); } sort(e+1,e+m+1); for (int i=1;i<=n;i++)fa[i]=i; for (int i=1;i<=m;i++) { if (e[i].z!=e[i-1].z) { a[++cnt].l=i; a[cnt-1].r=i-1; } int fx=getfa(e[i].x),fy=getfa(e[i].y); if (fx!=fy) { a[cnt].v++; fa[fx]=fy; tot++; } } a[cnt].r=m; if (tot!=n-1) { printf("0"); return 0; } for(int i=1;i<=n;i++)fa[i]=i; for(int i=1;i<=cnt;i++) { sum=0; dfs(i,a[i].l,0); ans=(ans*sum)%mod; for (int k=a[i].l;k<=a[i].r;k++) { int fx=getfa(e[k].x),fy=getfa(e[k].y); if (fx!=fy)fa[fx]=fy; } } printf("%d\n",ans); }
郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。