树上博弈(sg函数)

题目

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxn = 2e5 + 5, mod = 1e9 + 7;
vector<int> G[maxn];
int sg[maxn];//sg[i]表示以1为根时,以i为根的子树的sg值
int n, fa[maxn];
int sg2[maxn];//sg2[i]表示以i为根时,整棵树的sg值
void init(){
	for(int i = 1; i <= n; i++){
		sg[i] = sg2[i] = 0;
		G[i].clear();
	}
}
void dfs(int u, int f){
	fa[u] = f;
	for(auto v : G[u]){
		if(v == f) continue;
		dfs(v, u);
		sg[u] ^= (sg[v] + 1);//sg[u] = (sg[v1] + 1) ^ (sg[v2] + 1) ^ ... ^ (sg[vn] + 1)
	}
}
void dfs2(int u, int f){
	for(auto v : G[u]){
		if(v == f) continue;
		sg2[v] = sg[v] ^ ((sg2[u] ^ (sg[v] + 1)) + 1);//换根
		dfs2(v, u);
	}
}
int qpow(int a, int b){
	int res = 1;
	while(b){
		if(b & 1) res = res * a % mod;
		a = a * a % mod;
		b >>= 1;
	}
	return res;
}
void solve(){
	int i, j;
	cin >> n;
	init();
	for(i = 1; i < n; i++){
		int u, v;
		cin >> u >> v;
		G[u].push_back(v);
		G[v].push_back(u);
	}
  dfs(1, 1);
	int cnt = 0, SG;
	sg2[1] = sg[1];
	dfs2(1, 1);
	for(i = 1; i <= n; i++){
		if(sg2[i]) cnt++;
	}
  cout << cnt * qpow(n, mod - 2) % mod << '\n';
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	int T;
	cin >> T;
	while(T--){
		solve();
	}
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

__night_

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值