1016: [JSOI2008]最小生成树计数
Time Limit: 1 Sec Memory Limit: 162 MBSubmit: 5909 Solved: 2403
[ Submit][ Status][ Discuss]
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
1 2 1
1 3 1
1 4 1
2 3 2
2 4 1
3 4 1
Sample Output
8
#include<math.h>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
#define maxn 10005
#define mod 31011
int n,m,ans=1,sum,parent[maxn];
struct node
{
int u,v,c;
}a[maxn];
struct node1
{
int l,r,num;
}b[maxn];
bool comp(node a,node b)
{
return a.c<b.c;
}
int find(int x)
{
if(parent[x]==x)
return x;
return find(parent[x]);
}
void dfs(int x,int y,int num)
{
if(y==b[x].r+1)
{
if(num==b[x].num)
sum++;
return;
}
int t1=find(a[y].u);
int t2=find(a[y].v);
if(t1!=t2)
{
parent[t1]=t2;
dfs(x,y+1,num+1);
parent[t1]=t1;
}
dfs(x,y+1,num);
}
int main(void)
{
int i,j,cnt=0,tot=0;
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++)
scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].c);
sort(a+1,a+m+1,comp);
for(i=1;i<=n;i++)
parent[i]=i;
for(i=1;i<=m;i++)
{
int t1=find(a[i].u);
int t2=find(a[i].v);
if(a[i].c!=a[i-1].c)
{
b[++cnt].l=i;
b[cnt-1].r=i-1;
}
if(t1!=t2)
{
parent[t1]=t2;
b[cnt].num++;
tot++;
}
}
b[cnt].r=m;
if(tot!=n-1)
{
printf("0\n");
return 0;
}
for(i=1;i<=n;i++)
parent[i]=i;
for(i=1;i<=cnt;i++)
{
sum=0;
dfs(i,b[i].l,0);
ans=(ans*sum)%mod;
for(j=b[i].l;j<=b[i].r;j++)
{
int t1=find(a[j].u);
int t2=find(a[j].v);
if(t1!=t2)
parent[t1]=t2;
}
}
printf("%d\n",ans);
return 0;
}
#include<math.h>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
#define maxn 10005
#define mod 31011
int n,m,ans=1,sum,parent[maxn];
struct node
{
int u,v,c;
}a[maxn];
struct node1
{
int l,r,num;
}b[maxn];
bool comp(node a,node b)
{
return a.c<b.c;
}
int find(int x)
{
if(parent[x]==x)
return x;
return find(parent[x]);
}
void dfs(int x,int y,int num)
{
if(y==b[x].r+1)
{
if(num==b[x].num)
sum++;
return;
}
int t1=find(a[y].u);
int t2=find(a[y].v);
if(t1!=t2)
{
parent[t1]=t2;
dfs(x,y+1,num+1);
parent[t1]=t1;
}
dfs(x,y+1,num);
}
int main(void)
{
int i,j,cnt=0,tot=0;
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++)
scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].c);
sort(a+1,a+m+1,comp);
for(i=1;i<=n;i++)
parent[i]=i;
for(i=1;i<=m;i++)
{
int t1=find(a[i].u);
int t2=find(a[i].v);
if(a[i].c!=a[i-1].c)
{
b[++cnt].l=i;
b[cnt-1].r=i-1;
}
if(t1!=t2)
{
parent[t1]=t2;
b[cnt].num++;
tot++;
}
}
b[cnt].r=m;
if(tot!=n-1)
{
printf("0\n");
return 0;
}
for(i=1;i<=n;i++)
parent[i]=i;
for(i=1;i<=cnt;i++)
{
sum=0;
dfs(i,b[i].l,0);
ans=(ans*sum)%mod;
for(j=b[i].l;j<=b[i].r;j++)
{
int t1=find(a[j].u);
int t2=find(a[j].v);
if(t1!=t2)
parent[t1]=t2;
}
}
printf("%d\n",ans);
return 0;
}