#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
• 実行時間: 199 ms / 2,000 ms
• コード長: 980 bytes
• コンパイル時間: 4,370 ms
• コンパイル使用メモリ: 267,220 KB
• 実行使用メモリ: 25,856 KB
• 最終ジャッジ日時: 2024-12-15 02:59:59
• 合計ジャッジ時間: 7,428 ms
• ジャッジサーバーID （参考情報）: judge4 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	199 ms; 5,248 KB
testcase_02	AC	111 ms; 8,064 KB
testcase_03	AC	187 ms; 25,856 KB
testcase_04	AC	185 ms; 25,856 KB
testcase_05	AC	108 ms; 17,352 KB
testcase_06	AC	104 ms; 17,348 KB
testcase_07	AC	126 ms; 17,664 KB
testcase_08	AC	124 ms; 17,664 KB
testcase_09	AC	91 ms; 6,656 KB
testcase_10	AC	92 ms; 6,868 KB
testcase_11	AC	98 ms; 8,192 KB
testcase_12	AC	129 ms; 16,128 KB
testcase_13	AC	94 ms; 7,392 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();
	}
}