codeforces 1139c Edgy Trees 【并查集 】

题意：

一颗有n个节点的树

树的边为红色或者黑色

给你节点数 n  和 k

让你求出长度为k的 1 - n 的全排列 作为路径在该树上  经过黑色的边 的数量  答案 mod 1e9+7

题解：

将思路反过来想 我们先求出 所有的方案书  再减去不合格 的方案数量

想要让边都不为黑色 的路径  肯定在全是红色的边区域内  

这就是并查集  我们只要求出有多少块  并查集区域

然后在其中一块随机取得方案数减掉即可

并查集模板：

int find(int x){
	return x ==fin[x] ? x : fin[x] = find(fin[x]);
}
void Union(int u,int v){
	u = find(u);
	v = find(v);
	fin[u] = v;
}

	for(int i = 1; i <= n; i++) {
		num[find(i)]++;// find() 这是一个函数
	}

AC_code:

/*
Algorithm:
Author: anthony1314
Creat Time:
Time Complexity:
*/

#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<set>
#include<stack>
#include<cstring>
#include<cstdio>
//#include<bits/stdc++.h>
#define ll long long
#define maxn 100005
const ll mod = 1e9+7;
#define line printf("--------------");
using namespace std;
int fin[maxn], num[maxn], is[maxn];
ll quickpow(ll a, ll b) {
	ll ans = 1;
	while(b) {
		if(b & 1) {
			ans = ans * a % mod;
		}
		a = a * a % mod;
		b >>= 1;
	}
	return ans;
}
int find(int x){
	return x ==fin[x] ? x : fin[x] = find(fin[x]);
}
void Union(int u,int v){
	u = find(u);
	v = find(v);
	fin[u] = v;
}
int main() {
	memset(num, 0, sizeof(num));
	memset(is, 0, sizeof(is));
	int n, k;
	cin>>n>>k;
	for(int i = 1; i <= n; i++) {
		fin[i] = i;
	}
	int u, v, w;
	for(int i = 0; i < n-1; i++) {
		scanf("%d %d %d", &u, &v, &w);
		if(!w) {
			Union(u, v);
		}
	}
	for(int i = 1; i <= n; i++) {
		num[find(i)]++;
	}
	ll ans = quickpow(n, k);
	for(int i = 1; i <= n; i++) {
		if(num[i]) {
			ans -= quickpow(num[i], k);
			ans = (ans + mod) % mod;
		}
	}
	cout<<ans<<endl;
	return 0;
}