【bzoj 1016】[JSOI2008]最小生成树计数 脑残是病

拿到题以后我的首先得想法就是先建出一个最小生成树，然后对于现在最小生成树的每一条边都看看能不能替换（当有边和这一条边的边权值相同且连接的两端在并查集的不同集合就可以替换）最后利用乘法原理得出最后答案，但是细细思考其实是不成立的，因为会有重复使用的边，而且乘法原理会多次计算。但这却为正解有一个启发，就是每一种权值的个数出现的个数是一定的，这个其使用kruskal的算法实现就很显然。我们同样先做一次最小生成树，然后就可以得到每一种权值出现了多少次（离散化），又因为数据保证同样的权值不会出现10次，想到什么了，对就是状态枚举，枚举每种权值使用的情况然后O(n)的检查就可以得出每种权值的方法数由于不同权值之间互不影响，乘法原理搞定

脑残：1.离散化以后带入函数的权值还是用的原来的权值（调了好一会）

           2.在找一个权值的边的起点的时候因为懒直接用建树时候的，最后无赖写了一个二分，哎，其实大可不必只是  一开始想偷懒

#include<cstdio>
#include<cstring>
#include<istream>
#define maxn 5020
#define PB make_pair
#include<vector>
#include<algorithm>
#include<cstdlib>
#define LL long long
using namespace std;
const int Mod=31011;
int n,m,cur[maxn],len[maxn],fa[maxn],tot,yes[maxn],edg[maxn],much[maxn];
struct edge{
	int a,b,w;
	bool operator <(const edge& b)const{return w<b.w;}
	edge():a(0),b(0),w(0){}
}e[maxn];

void read(int& x){
	char c=getchar();
	x=0;
	for(;c<'0'||c>'9';c=getchar());
	for(;c>='0'&&c<='9';c=getchar()){
		x=x*10+c-'0';
	}
}
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);}
int count(int x){return x==0?0:count(x/2)+(x&1);}
bool unio(int x,int y){
	int dx=find(x),dy=find(y);
	if(dx==dy)return false;
	fa[dx]=dy;
	return true;
}

bool  build(){
	for(int i=1;i<=n;i++)fa[i]=i;
	int cnt=0;
	for(int i=1;i<=m;i++){
		int x=e[i].a,y=e[i].b,w=e[i].w;
		much[w]++;
		if(!unio(x,y))continue;
		len[w]++;		
		edg[++cnt]=i;
	}
	if(cnt!=n-1){
		printf("0");
		return false;
	}
	return true;
}

int finde(int x){
	int l=1,r=m;
	while(l<r){
		int mid=l+r>>1;
		if(e[mid].w<x)l=mid+1;
		else r=mid;
	}
	return l;
}

bool rebuild(int pos,int l){
	for(int i=1;i<=n;i++)fa[i]=i;
	int cnt=0,begin;

	for(int i=1;i<n;i++){
		int x=e[edg[i]].a,y=e[edg[i]].b,w=e[edg[i]].w;
		if(w==l)continue;
		unio(x,y);
		cnt++;
	}
	begin=finde(l);
	for(int i=0;i<much[l];i++){ 
		if(pos&(1<<i)){
			if(unio(e[begin+i].a,e[begin+i].b))cnt++;
		}
	}
	if(cnt==n-1)return true;
	return false ;
}

int solve(int x){
	if(len[x]==0)return 1;
	int cnt=len[x];
	int p=0;
	for(int i=0;i<(1<<much[x]);i++){
		if(count(i)!=cnt)continue;
		if(rebuild(i,x))p++;
		p%=Mod;
	}
	return p%Mod;
}

int main(){
	read(n),read(m);
	for(int a,b,c,i=1;i<=m;i++){
		read(a),read(b),read(c);
		e[i].a=a,e[i].b=b,e[i].w=c;
		cur[i]=c;
	}
	
	sort(cur+1,cur+1+m);
	tot=unique(cur+1,cur+1+m)-cur-1;
	for(int i=1;i<=m;i++){
		e[i].w=lower_bound(cur+1,cur+1+tot,e[i].w)-cur;
	}
	sort(e+1,e+1+m);
	if(!build())return 0;
	for(int i=1;i<=tot;i++){
		yes[i]=solve(i)%Mod;
	}
	int ANS=1;
	for(int i=1;i<=tot;i++)ANS=(LL)ANS*yes[i]%Mod;
	printf("%d",ANS%Mod);
	return 0;
}