#971913 (C++17) No.2726 Rooted Tree Nim

• 問題: No.2726 Rooted Tree Nim
• ユーザー: inksamurai
• 提出日時: 2024-04-12 23:17:15
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 196 ms / 2,000 ms
• コード長: 1,123 bytes
• コンパイル時間: 2,212 ms
• コンパイル使用メモリ: 207,728 KB
• 実行使用メモリ: 25,984 KB
• 最終ジャッジ日時: 2024-10-02 23:47:20
• 合計ジャッジ時間: 4,898 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	69 ms; 5,248 KB
testcase_02	AC	107 ms; 8,832 KB
testcase_03	AC	196 ms; 25,984 KB
testcase_04	AC	183 ms; 25,984 KB
testcase_05	AC	103 ms; 18,744 KB
testcase_06	AC	103 ms; 18,748 KB
testcase_07	AC	131 ms; 19,584 KB
testcase_08	AC	124 ms; 19,712 KB
testcase_09	AC	86 ms; 7,168 KB
testcase_10	AC	87 ms; 7,320 KB
testcase_11	AC	90 ms; 8,704 KB
testcase_12	AC	120 ms; 17,712 KB
testcase_13	AC	89 ms; 7,240 KB

ソースコード

#include <bits/stdc++.h>
#define int ll
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define rng(i,c,n) for(int i=c;i<n;i++)
#define fi first
#define se second
#define pb push_back
#define all(a) a.begin(), a.end()
#define sz(a) ((int) a.size())
#define vec(...) vector<__VA_ARGS__>
#define _49vEjP0 ios::sync_with_stdio(0),cin.tie(0);
typedef long long ll;
typedef vector<int> vi;
typedef pair<int,int> pii;
void print(){cout<<'\n';}
template<class h,class...t>
void print(const h&v,const t&...u){cout<<v<<' ',print(u...);}

void slv(){
	int n,k;
	cin>>n>>k;

	vec(vi) adj(n);
	rep(i,n-1){
		int u,v;
		cin>>u>>v;
		u-=1,v-=1;
		adj[u].pb(v),adj[v].pb(u);
	}

	vi a(n);
	rep(i,n){
		cin>>a[i];
	}

	vi dep(n);
	auto dfs=[&](auto self,int v,int par)->void{
		for(auto u:adj[v]){
			if(u==par) continue;
			dep[u]=dep[v]+1;
			self(self,u,v);
		}
	};
	dfs(dfs,0,-1);

	int sxor=0;
	rep(v,n){
		if(dep[v]%2){
			sxor^=(a[v]%(k+1));
		}
	}
	if(sxor==0){
		cout<<"P\n";
	}else{
		cout<<"K\n";
	}
}

signed main(){
_49vEjP0;
	int __t;
	cin>>__t;
	rep(cs,__t){
		slv();
	}
}