結果

問題 No.2726 Rooted Tree Nim
ユーザー ttkkggwwttkkggww
提出日時 2024-05-08 19:35:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 962 bytes
コンパイル時間 4,805 ms
コンパイル使用メモリ 266,324 KB
実行使用メモリ 25,960 KB
最終ジャッジ日時 2024-05-08 19:35:13
合計ジャッジ時間 8,528 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 197 ms
6,940 KB
testcase_02 WA -
testcase_03 AC 153 ms
25,960 KB
testcase_04 WA -
testcase_05 AC 93 ms
17,356 KB
testcase_06 AC 88 ms
17,356 KB
testcase_07 AC 105 ms
17,716 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
権限があれば一括ダウンロードができます

ソースコード

diff #

#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;
	}
	for(int i = 0;i<N;i++){
		if(depth[i]%2==1){
			depthxor[depth[i]] ^= A[i];
		}
	}
	sort(depthxor.begin(),depthxor.end());
	if(depthxor.back()==0){
		cout<<"P"<<endl;
	}else{
		cout<<"K"<<endl;
	}
}

signed main(){
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	int T;
	cin >> T;
	while(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];
		solve();
	}
}
0