bzoj1016: [JSOI2008]最小生成树计数（kruskal+dfs）

　　一直以为这题要martix-tree，实际上因为有相同权值的边不大于10条于是dfs就好了...

　　先用kruskal求出每种权值的边要选的次数num，然后对于每种权值的边2^num暴搜一下选择的情况算出多少种情况合法，对于每种权值的边的方案用乘法原理乘起来就是答案了

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=,mod=;
struct poi{int x,y,z,pos;}a[maxn];
int n,m,cntt,sum,cnt,ans;
int num[maxn],fa[maxn],l[maxn],r[maxn];
inline void read(int &k)
{
    int f=;k=;char c=getchar();
    while(c<''||c>'')c=='-'&&(f=-),c=getchar();
    while(c<=''&&c>='')k=k*+c-'',c=getchar();
    k*=f;
}
bool cmp(poi a,poi b){return a.z<b.z;}
int gf(int x){return fa[x]==x?x:gf(fa[x]);}
void dfs(int w,int x,int dep)
{
    if(dep>num[w])return;
    if(x>r[w]){if(dep==num[w])sum++;return;}
    int xx=gf(a[x].x),yy=gf(a[x].y);
    if(xx!=yy)fa[xx]=yy,dfs(w,x+,dep+),fa[xx]=xx;
    dfs(w,x+,dep);
}
int main()
{
    read(n);read(m);
    for(int i=;i<=m;i++)read(a[i].x),read(a[i].y),read(a[i].z);
    sort(a+,a++m,cmp);
    for(int i=;i<=m;i++)
    {
        if(a[i].z!=a[i-].z)r[cnt++]=i-,l[cnt]=i;
        a[i].pos=cnt;
    }
    r[cnt]=m;
    for(int i=;i<=n;i++)fa[i]=i;
    for(int i=;i<=m;i++)
    {
        int x=gf(a[i].x),y=gf(a[i].y);
        if(x!=y)fa[x]=y,num[a[i].pos]++,cntt++;
    }
    if(cntt!=n-)return puts(""),;
    for(int i=;i<=n;i++)fa[i]=i;ans=;
    for(int i=;i<=cnt;i++)
    if(num[i])
    {
        sum=;dfs(i,l[i],);
        for(int j=l[i];j<=r[i];j++)fa[gf(a[j].x)]=gf(a[j].y);
        ans=1ll*ans*sum%mod;
    }
    printf("%d\n",ans);
}