最小生成树计数
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
4 6
1 2 1
1 3 1
1 4 1
2 3 2
2 4 1
3 4 1
Sample Output
8
对着题目撕烤了好久都没想出来
直到在某题解的指导下发现相同边权的边不超过10……
思路:
考虑到边权相同的边数不超过10,同时n、m这么小,怕不是指数级算法……
于是,先做一遍最小生成树,统计出每种边权最小生成树上有几条。
然后就可以按照边权分开考虑了。
把边权相同的边放到一起,枚举选择方案并判断是否成环,如果成环则此方案不合法,否则当前边权的方案数+1。
统计完一种边权后,把当前边权的方案数乘入总方案数中,然后把这些边的影响消去——也就是把这些边所联通的联通块缩点。
然后继续做下一个边权的边,即可~
#include<bits/stdc++.h>
using namespace std;
inline int read()
{
int x=0;char ch=getchar();
while(ch<'0' || '9'<ch)ch=getchar();
while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
return x;
}
typedef long long ll;
const int N=109;
const int M=1009;
const int K=(1<<12)-9;
const int md=31011;
struct ed
{
int u,v,w;
bool operator < (ed o)const{return w<o.w;}
}e[M];
int n,m,ans;
int fa[N],bitcnt[K];
map<int,int> ha;
inline int find(int x){return fa[x]=((fa[x]==x || !fa[x])?x:find(fa[x]));}
inline void init()
{
for(int i=1;i<K;i++)
bitcnt[i]=bitcnt[i>>1]+(i&1);
}
inline void kruscal()
{
int cnt=1;
for(int i=1;i<=m && cnt<n;i++)
{
int u=find(e[i].u),v=find(e[i].v);
if(u==v)continue;
fa[u]=v;cnt++;
ha[e[i].w]++;
}
if(cnt!=n)
puts("0"),exit(0);
}
inline int work(int l)
{
int r,cnt=ha[e[l].w],tans=0;
for(r=l;r<m && e[r].w==e[r+1].w;r++);
for(int i=0,ei=1<<r-l+1;i<ei;i++)
if(bitcnt[i]==cnt)
{
memset(fa,0,sizeof(fa));
for(int j=l;j<=r;j++)
if((i>>j-l)&1)
{
int u=find(e[j].u),v=find(e[j].v);
if(u==v)goto nxt;
fa[u]=v;
}
tans++;
nxt:;
}
ans=ans*tans%md;
memset(fa,0,sizeof(fa));
for(int i=l;i<=r;i++)
{
int u=find(e[i].u),v=find(e[i].v);
if(u==v)continue;
fa[u]=v;
}
for(int i=r+1;i<=m;i++)
e[i].u=find(e[i].u),e[i].v=find(e[i].v);
return r;
}
int main()
{
init();
n=read();m=read();
for(int i=1;i<=m;i++)
e[i].u=read(),e[i].v=read(),e[i].w=read();
sort(e+1,e+m+1);
kruscal();ans=1;
for(int i=1;i<=m;i++)
if(ha.find(e[i].w)!=ha.end())
i=work(i);
printf("%d\n",ans);
return 0;
}