[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

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;
}