bzoj1016 [JSOI2008]最小生成树计数

先做一遍kruskal，然后发现不同的方案只可能是相同权值的不同的边干了相同的事

题目又保证了相同权值的边数很少，直接状压即可

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#define N 1005
using namespace std;
int getnum(int x){
    int cnt=0;
    while(x){cnt+=x&1;x>>=1;}
    return cnt;
}
int n,m,ans=1;
int fa[N],tmp[N];
int find(int x,int *y){
    if(x==y[x])return x;
    return y[x]=find(y[x],y);
}
struct edge{int u,v,w;}ed[10*N];
bool cmp(edge a,edge b){return a.w<b.w;}
bool vis[10*N];
int bit[15];
int main(){
    bit[0]=1;
    for(int i=1;i<=10;i++)bit[i]=bit[i-1]<<1;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&ed[i].u,&ed[i].v,&ed[i].w);
    sort(ed+1,ed+m+1,cmp);
    for(int i=1,fv,fu;i<=m;i++){
        fu=find(ed[i].u,fa);fv=find(ed[i].v,fa);
        if(fu!=fv){
            fa[fu]=fv;
            vis[i]=1;
        }
    }
    for(int be=1,en,num,len,cnt,flag;be<=m;be++){
        for(en=be;en<=m&&ed[en+1].w==ed[be].w;en++);
        for(int i=1;i<=n;i++)fa[i]=tmp[i]=i;
        num=n-1;cnt=0;
        for(int i=1;i<=m;i++){
            if(i>=be&&i<=en)continue;
            if(vis[i]){
                int fu=find(ed[i].u,tmp),fv=find(ed[i].v,tmp);
                tmp[fu]=fv;num--;
            }
        }
        len=en-be+1;
        for(int i=0;i<bit[len];i++)if(getnum(i)==num){
            flag=0;
            memcpy(fa,tmp,sizeof fa);
            for(int j=be;j<=en;j++){
                if(i&bit[j-be]){
                    int fu=find(ed[j].u,fa),fv=find(ed[j].v,fa);
                    if(fu==fv){flag=1;break;}
                    fa[fu]=fv;
                }
            }
            if(flag==0)cnt++;
        }
        ans=ans*cnt%31011;
        be=en;
    }
    printf("%d
",ans);
    return 0;
}