C. Bakry and Partitioning（codeforces）

思路：

假设在不删除边的情况下，a1^a2^.......^an=sum。

基础铺垫：一个数x，x^x=0,  x^x^x=x。

1、如果sum=0，那么可以仅删除一条边，把一个连通块分为两个异或和相同的连通块。

2、如果sum!=0，k=2的话，绝对不成立，至少删除一条边，至多删除一条边，和第一种情况冲突。

我们考虑三个及以上个连通块，根据铺垫知识可以知道，每三个异或和相同的连通块可以看作一个连通块。因此最终我们可以只考虑能否形成3个权值异或和相等，并且权值异或和为sum的连通块即可，即删两条边。这个操作可以通过dfs当前结点和其子树异或和来得到。如果递归到某个结点其与其所有子节点的异或和为sum那么可割点加1，返回0，否则返回当前结点和其子结点的异或和即可。

代码：

#include<bits/stdc++.h>
#define int long long
#define x first
#define y second
#define endl '\n'
#define pq priority_queue
using namespace std;
typedef pair<int,int> pii;
vector<int>e[200010];
int a[200010];
int ans,sum;

int dfs(int u,int p){
	int t=a[u];
	for(auto v:e[u]){
		if(v==p)continue;
		t^=dfs(v,u);
	}
	if(sum==t){
		t=0,ans++;
	}
	return t;
}

void solve(){
	int n,k;cin>>n>>k;
	sum=0;
	ans=0;
	for(int i=1;i<=n;i++)
		e[i].clear();
	for(int i=1;i<=n;i++)
		cin>>a[i],sum^=a[i];
	for(int i=1;i<n;i++){
		int x ,y;cin>>x>>y;
		e[x].push_back(y);
		e[y].push_back(x);
	}
	if(sum==0){
		cout<<"YES"<<endl;return;
	}
	if(k==2){
		cout<<"NO"<<endl;return;
	}
	dfs(1,-1);
	if(ans<2)cout<<"NO"<<endl;
	else cout<<"YES"<<endl;
}

signed main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    int t;cin>>t;
    while(t--)
    {
        solve();
    }
    return 0;
}