jsoi2008最小生成树计数bzoj1016

嘛，就是把一组权值相等的边分成一组，然后对于每个连通块暴力一下生成树个数，然后用并查集做一下暴力的缩点就好了，乘法原理即可，注意不能写路径压缩……手贱打了一个结果连样例都过不了去掉后就A了ＱＡＱＱＱＱＱＱＱＱ

代码片怎么搞完全布吉岛，算了还是直接贴代码

#include<cstdio>
#include<algorithm>
#define N 105
#define M 1005
#define inf 31011
using namespace std;
struct edge
{
        int x,y,w;
}e[M];
struct wakaka
{
        int l,r,num;
}a[M];
bool compare(edge a,edge b)
{
        return a.w<b.w;
}
int f[N];
int findd(int x)
{
        if (f[x]==x) return x;
        else return findd(f[x]);
}
int ans=1,n,m,tot,cnt,sum;
void dfs(int k,int p,int q)
{
        if (p==a[k].r+1)
        {
                if (q==a[k].num) sum++;
                return;
        }
        if (q>a[k].num) return;
        int x=e[p].x,y=e[p].y;
        x=findd(x);y=findd(y);
        if (x!=y)
        {
                f[x]=y;
                dfs(k,p+1,q+1);
                f[x]=x;f[y]=y;
        }
        dfs(k,p+1,q);
}

int main()
{
        scanf("%d%d",&n,&m);
        for (int i=1;i<=m;i++)
                scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].w);
        for (int i=1;i<=n;i++) f[i]=i;
        sort(e+1,e+m+1,compare);
        for(int i=1;i<=m;i++)
        {
                if(e[i].w!=e[i-1].w)
                {
                        a[++cnt].l=i;
                        a[cnt-1].r=i-1;
                }
                int p=findd(e[i].x),q=findd(e[i].y);
                if(p!=q) {f[p]=q;a[cnt].num++;tot++;}
        }
        a[cnt].r=m;
        if (tot!=n-1) {printf("0"); return 0;}
        for (int i=1;i<=m;i++) f[i]=i;
        for (int i=1;i<=cnt;i++)
        {
                sum=0;
                dfs(i,a[i].l,0);
                ans=(ans*sum)%inf;
                for (int j=a[i].l;j<=a[i].r;j++)
                {
                        int p=findd(e[j].x),q=findd(e[j].y);
                        if (p!=q) f[p]=q;
                }
        }
        printf("%d\n",ans);
        return 0;
}