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

• 問題: No.2726 Rooted Tree Nim
• ユーザー: ttkkggww
• 提出日時: 2024-05-08 19:56:47
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 210 ms / 2,000 ms
• コード長: 980 bytes
• コンパイル時間: 5,042 ms
• コンパイル使用メモリ: 265,560 KB
• 実行使用メモリ: 25,900 KB
• 最終ジャッジ日時: 2024-05-08 19:56:57
• 合計ジャッジ時間: 7,855 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 6,812 KB
testcase_01	AC	203 ms; 6,812 KB
testcase_02	AC	117 ms; 8,064 KB
testcase_03	AC	202 ms; 25,836 KB
testcase_04	AC	210 ms; 25,900 KB
testcase_05	AC	112 ms; 17,344 KB
testcase_06	AC	110 ms; 17,356 KB
testcase_07	AC	132 ms; 17,664 KB
testcase_08	AC	135 ms; 17,532 KB
testcase_09	AC	94 ms; 6,912 KB
testcase_10	AC	94 ms; 6,872 KB
testcase_11	AC	101 ms; 8,192 KB
testcase_12	AC	146 ms; 16,000 KB
testcase_13	AC	100 ms; 7,388 KB

ソースコード

#include<bits/stdc++.h>
#include <iostream>
using namespace std;
#include<atcoder/all>
using namespace atcoder;
using ll = long long;
int N,K;
vector<vector<int>> G;
vector<int> A;
vector<int> depth;

void dfs(int v,int p,int d){
	depth[v] = d;
	for(int u:G[v]){
		if(u==p) continue;
		dfs(u,v,d+1);
	}
}

void solve(){
	depth = vector<int>(N,0);
	dfs(0,-1,0);
	vector<int> depthxor(N);
	for(int i = 0;i<N;i++){
		A[i] %= K+1;
	}
	int x = 0;
	for(int i = 0;i<N;i++){
		if(depth[i]%2==1){
			x ^= A[i];
		}
	}
	sort(depthxor.begin(),depthxor.end());
	if(x==0){
		cout<<"P"<<endl;
	}else{
		cout<<"K"<<endl;
	}
}

signed main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	int T;
	cin >> T;
	for(int t= 0;t<T;t++){
		cin >> N >> K;
		G = vector<vector<int>>(N);
		A.resize(N);
		for(int i = 0;i<N-1;i++){
			int u,v;
			cin >> u >> v;
			u--; v--;
			G[u].push_back(v);
			G[v].push_back(u);
		}
		for(int i = 0;i<N;i++) cin >> A[i];
		if(t==23)cerr<<N<<endl;
		solve();
	}
}