結果

問題 No.2726 Rooted Tree Nim
ユーザー DaylightDaylight
提出日時 2024-04-12 23:00:53
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 217 ms / 2,000 ms
コード長 3,322 bytes
コンパイル時間 2,582 ms
コンパイル使用メモリ 203,112 KB
実行使用メモリ 27,472 KB
最終ジャッジ日時 2024-04-12 23:01:00
合計ジャッジ時間 6,056 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 217 ms
6,940 KB
testcase_02 AC 121 ms
9,216 KB
testcase_03 AC 216 ms
27,404 KB
testcase_04 AC 210 ms
27,472 KB
testcase_05 AC 138 ms
18,108 KB
testcase_06 AC 147 ms
18,108 KB
testcase_07 AC 145 ms
19,648 KB
testcase_08 AC 140 ms
19,484 KB
testcase_09 AC 95 ms
7,296 KB
testcase_10 AC 97 ms
7,072 KB
testcase_11 AC 101 ms
8,832 KB
testcase_12 AC 142 ms
17,756 KB
testcase_13 AC 104 ms
7,280 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;
#define SZ(x) (int) (x).size()
#define REP(i, n) for(int i = 0; i < (n); i++)
#define FOR(i, a, b) for(auto i = (a); i < (b); i++)
#define For(i, a, b, c) \
	for(auto i = (a); i != (b); i += (c))
#define REPR(i, n) for(auto i = (n) -1; i >= 0; i--)
#define ALL(s) (s).begin(), (s).end()
#define so(V) sort(ALL(V))
#define rev(V) reverse(ALL(V))
#define uni(v) v.erase(unique(ALL(v)), (v).end())
#define eb emplace_back

typedef unsigned long long ull;
typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<bool> vb;
typedef vector<vi> vvi;
typedef vector<vll> vvll;
typedef pair<int, int> PI;
typedef pair<ll, ll> PL;
const double EPS = 1e-6;
const int MOD = 1e9 + 7;
const int INF = (1 << 30);
const ll LINF = 1e18;
const double math_PI = acos(-1);

template<typename T>
vector<T> make_v(size_t a) {
	return vector<T>(a);
}

template<typename T, typename... Ts>
auto make_v(size_t a, Ts... ts) {
	return vector<decltype(make_v<T>(ts...))>(
		a, make_v<T>(ts...));
}

template<typename T, typename V>
typename enable_if<is_class<T>::value == 0>::type fill_v(
	T& t, const V& v) {
	t = v;
}

template<typename T, typename V>
typename enable_if<is_class<T>::value != 0>::type fill_v(
	T& t, const V& v) {
	for(auto& e: t) fill_v(e, v);
}

template<class T>
bool chmax(T& a, const T& b) {
	if(a < b) {
		a = b;
		return true;
	}
	return false;
}

template<class T>
bool chmin(T& a, const T& b) {
	if(a > b) {
		a = b;
		return true;
	}
	return false;
}

template<typename S, typename T>
istream& operator>>(istream& is, pair<S, T>& p) {
	cin >> p.first >> p.second;
	return is;
}

template<typename T>
istream& operator>>(istream& is, vector<T>& vec) {
	for(T& x: vec) is >> x;
	return is;
}

template<typename T>
string join(vector<T>& vec, string splitter) {
	stringstream ss;
	REP(i, SZ(vec)) {
		if(i != 0) ss << splitter;
		ss << vec[i];
	}
	return ss.str();
}

template<typename T>
ostream& operator<<(ostream& os, vector<T>& vec) {
	os << join(vec, " ");
	return os;
}

#ifdef LOCAL
#include "./cpp-dump/dump.hpp"
#include "./cpp-dump/mytypes.hpp"
#define dump(...) cpp_dump(__VA_ARGS__)
namespace cp = cpp_dump;
#else
#define dump(...)
#define preprocess(...)
#define CPP_DUMP_SET_OPTION(...)
#define CPP_DUMP_DEFINE_EXPORT_OBJECT(...)
#define CPP_DUMP_DEFINE_EXPORT_ENUM(...)
#define CPP_DUMP_DEFINE_DANGEROUS_EXPORT_OBJECT(...)
#endif
template<class T = ll>
struct Edge {
public:
	int from, to;
	T cost;
	Edge() {
	}
	Edge(int _from, int _to, T _cost) {
		from = _from;
		to = _to;
		cost = _cost;
	}
};
template<class T = ll>
using Edges = vector<Edge<T>>;
template<class T = ll>
using Graph = vector<Edges<T>>;

void solve() {
	int N, K;
	cin >> N >> K;
	Graph<ll> G(N);
	REP(i, N - 1) {
		int X, Y;
		cin >> X >> Y;
		X--;
		Y--;
		G[X].eb(X, Y, 1);
		G[Y].eb(Y, X, 1);
	}
	vi A(N);
	cin >> A;
	ll X = 0;
	auto dfs = [&](auto f, int c = 0, int p = -1,
				   int d = 0) -> void {
		if(d % 2 == 1) {
			X ^= A[c] % (K + 1);
		}
		for(auto e: G[c]) {
			if(e.to == p) continue;
			f(f, e.to, c, d + 1);
		}
	};
	dfs(dfs);
	if(X == 0) {
		cout << "P" << endl;
	} else {
		cout << "K" << endl;
	}
}

int main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);
	preprocess();
	int T;
	cin >> T;
	while(T--) {
		solve();
	}
	return 0;
}
0