【XSY2716】营养餐 博弈论

题目描述

　　给你一棵有根树，每个点有两个属性\(a,b\)

　　两人轮流操作，每次要减小一个点的\(a\)值，要求

\[a_x\geq\sum_{i\in child(x)}a_ib_i \]

　　保证初始状态满足这个要求。

　　\(\sum n\leq 5\times {10}^5\)

题解

　　令

\[s_x=a_x-\sum_{i\in child(x)}a_ib_i \]

　　每次操作相当于减小\(s_x\)，把\(s_{f_x}\)加上减小的值$\times $$b_x$。

　　当\(b_x=0\)时\(x\)对\(f_x\)没有影响，可以把\(x\)视为根。

　　把原树划分成森林后做阶梯博弈即可。

　　计算出所有深度为\(x\)的点的\(s_x\)异或和，如果非零则先手胜，否则后手胜。

　　阶梯博弈：所有深度为偶数的点的信息是没有用的。如果把某一个偶数层的点的值挪到奇数层的点上，对手可以再把这些值挪到偶数层的点上。所以最好情况都不会对自己有利，就不会这么决策。

　　时间复杂度：\(O(n)\)

代码

#include<cstdio>
#include<cstring>
using namespace std;
struct graph
{
	int v[100010];
	int t[100010];
	int h[50010];
	int n;
	void add(int x,int y)
	{
		n++;
		v[n]=y;
		t[n]=h[x];
		h[x]=n;
	}
	void init()
	{
		memset(h,0,sizeof h);
		n=0;
	}
};
graph g;
int f[100010];
int ban[100010];
int s[100010];
int a[100010];
int b[100010];
void dfs(int x,int fa)
{
	f[x]=fa;
	s[x]=a[x];
	int i;
	for(i=g.h[x];i;i=g.t[i])
		if(g.v[i]!=fa)
		{
			dfs(g.v[i],x);
			s[x]-=a[g.v[i]]*b[g.v[i]];
		}
}
int ans;
void dfs2(int x,int d)
{
	if((d&1))
		ans^=s[x];
	int i;
	for(i=g.h[x];i;i=g.t[i])
		if(g.v[i]!=f[x]&&!ban[g.v[i]])
			dfs2(g.v[i],d+1);
}
void solve()
{
	int n;
	scanf("%d",&n);
	int i,x,y;
	for(i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(i=1;i<=n;i++)
		scanf("%d",&b[i]);
	g.init();
	for(i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		g.add(x,y);
		g.add(y,x);
	}
	for(i=1;i<=n;i++)
		ban[i]=0;
	for(i=1;i<=n;i++)
		if(!b[i]||i==1)
			ban[i]=1;
	dfs(1,0);
	ans=0;
	for(i=1;i<=n;i++)
		if(ban[i])
			dfs2(i,1);
	if(ans)
		printf("YES\n");
	else
		printf("NO\n");
}
int main()
{
#ifndef ONLINE_JUDGE
	freopen("c.in","r",stdin);
	freopen("c.out","w",stdout);
#endif
	int t;
	scanf("%d",&t);
	while(t--)
		solve();
	return 0;
}