[cf] 746 div2 C. Bakry and Partitioning

前言

$Tag:$异或 树 dfs *1700
传送门 :

题意 :
给定一棵树,你可以删除最少$1$,对多$k-1$ 条边,询问是否可以使得删除边之后,每个连通分量中的异或和相等

思路 :

考虑删除一条边($k=2$)的情况

$a_x\oplus (a_1\oplus a_2....a_n)=0$

也就是$a_1\oplus a_2\oplus a_3.....a_n=sum=0$

如果$sum!=0$,那么因为两个连通块需要异或和相同$b\oplus b=sum=0$,但是$sum!=0$

因此$k=2$的时候,只有$sum=0$是合法的

然后我们考虑删除多条边的情况$k>=3$

---

首先这里抛出一个贪心的思路 :
在异或考虑最优的情况下 :
如果选择 三个商品异或相同 并且 五个商品异或也相同,那么我们考虑选择三个商品的最优
up:电音不抖腿

---

因此我们可以$dfs$每一个子树,然后计算其异或和,当且仅当第一次$total==sum$的时候,我们就进行一次分割

同时又因为$a\oplus a\oplus a=a$因此最少都需要两个割点,所以最后判断分割的点是否是两个以上即可

code :

vector<int> g[N];
int n,k;
int w[N];
int cnt;

int total;

int dfs(int u,int fa){
	int val =  w[u];
	
	for(auto x :  g[u]){
		if(x == fa)continue;
		
		int tmp = dfs(x,u);
		if(tmp == total) cnt ++;
		else val^=tmp;
		
	}
	return val;
	
}
void solve(){
	cin>>n>>k;
	total = 0 ;
	cnt = 0;
	
	
	for(int i = 1;i<=n; i++ )g[i].clear();
	for(int i = 1;i<=n; i++ ){
		cin>>w[i];
		total^=w[i];
	}
	
	for(int i=1;i<n; i++){
		int u,v;cin>>u>>v;
		g[u].pb(v);
		g[v].pb(u);
	}
	
	if(total == 0){
		cout<<"YES"<<endl;
		return;
	}
	
	if(k<=2){
		cout<<"NO"<<endl;
		return;
	}
	
	
	dfs(1,-1);
	if(cnt > 1){
		cout<<"YES"<<endl;
		return;
	}
	
	cout<<"NO"<<endl;
	
}

int main(){
    int t;cin>>t;while(t--)
    solve();
    return 0 ;
}