C. Edgy Trees

链接

[https://codeforces.com/contest/1139/problem/C]

题意

给你n个点，n-1个边，无向的。有red和black的。
k表示经过这k个点。可以跨其他点

分析

先算所有的，再减去不符合即可以了。不符合就是都走0的那种
用dfs求联通块并记录该块有多少个点就可以了，看代码吧

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=2e5+10;
const ll mod=1e9+7;
ll po(ll x,ll y){
	ll ba=x,ans=1;
	while(y){
		if(y&1) ans=(ans*ba)%mod;
		ba=(ba*ba)%mod;
		 y>>=1;
	}
	return ans;
}
vector<int> ve[N];
bool vis[N];
ll cnt;
void dfs(int i){
	if(vis[i]) return;
	vis[i]=1;
	cnt++;
	for(int j=0;j<ve[i].size();j++)
	dfs(ve[i][j]);
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	//freopen("in.txt","r",stdin);
	ll n,k;
        cin>>n>>k;
		int u,v,x;
		memset(vis,0,sizeof(vis));
		for(int i=1;i<n;i++){
			cin>>u>>v>>x;
			if(x==0)
			ve[u].push_back(v),ve[v].push_back(u);
		}
		ll sum=po(n,k);
		for(int i=1;i<=n;i++)
		{
			cnt=0;
			if(!vis[i]) 
			{
				dfs(i);
			  sum-=po(cnt,k);
			  sum+=mod;
			  sum%=mod; 
			}
		}
		cout<<sum<<endl;
	return 0;
}