結果
問題 | No.2726 Rooted Tree Nim |
ユーザー | 👑 amentorimaru |
提出日時 | 2024-04-12 22:49:11 |
言語 | C++23 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 270 ms / 2,000 ms |
コード長 | 3,813 bytes |
コンパイル時間 | 4,182 ms |
コンパイル使用メモリ | 231,140 KB |
実行使用メモリ | 25,088 KB |
最終ジャッジ日時 | 2024-10-02 23:38:18 |
合計ジャッジ時間 | 7,239 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 270 ms
5,248 KB |
testcase_02 | AC | 212 ms
8,192 KB |
testcase_03 | AC | 242 ms
25,088 KB |
testcase_04 | AC | 246 ms
25,088 KB |
testcase_05 | AC | 182 ms
17,444 KB |
testcase_06 | AC | 183 ms
17,312 KB |
testcase_07 | AC | 215 ms
18,112 KB |
testcase_08 | AC | 220 ms
18,176 KB |
testcase_09 | AC | 188 ms
7,072 KB |
testcase_10 | AC | 189 ms
7,092 KB |
testcase_11 | AC | 190 ms
8,448 KB |
testcase_12 | AC | 212 ms
16,512 KB |
testcase_13 | AC | 195 ms
7,244 KB |
ソースコード
#define ATCODER #define _USE_MATH_DEFINES #include<stdio.h> #include<iostream> #include<fstream> #include<algorithm> #include<vector> #include<string> #include <cassert> #include <numeric> #include <unordered_map> #include <unordered_set> #include <queue> #include <math.h> #include <climits> #include <set> #include <map> #include <list> #include <random> #include <iterator> #include <bitset> #include <chrono> #include <type_traits> using namespace std; using ll = long long; using ld = long double; using pll = pair<ll, ll>; using pdd = pair<ld, ld>; //template<claLR T> using pq = priority_queue<T, vector<T>, greater<T>>; #define FOR(i, a, b) for(ll i=(a); i<(b);i++) #define REP(i, n) for(ll i=0; i<(n);i++) #define ROF(i, a, b) for(ll i=(b-1); i>=(a);i--) #define PER(i, n) for(ll i=n-1; i>=0;i--) #define REPREP(i,j,a,b) for(ll i=0;i<a;i++)for(ll j=0;j<b;j++) #define VV(type) vector< vector<type> > #define VV2(type,n,m,val) vector< vector<type> > val;val.resize(n);for(ll i;i<n;i++)val[i].resize(m) #define vec(type) vector<type> #define VEC(type,n,val) vector<type> val;val.resize(n) #define VL vector<ll> #define VVL vector< vector<ll> > #define VP vector< pair<ll,ll> > #define SZ size() #define all(i) begin(i),end(i) #define SORT(i) sort(all(i)) #define BITI(i) (1<<i) #define BITSET(x,i) x | (ll(1)<<i) #define BITCUT(x,i) x & ~(ll(1)<<i) #define EXISTBIT(x,i) (((x>>i) & 1) != 0) #define ALLBIT(n) (ll(1)<<n-1) #define CHMAX(n,v) n=n<v?v:n #define CHMIN(n,v) n=n>v?v:n #define MP(a,b) make_pair(a,b) #define DET2(x1,y1,x2,y2) (x1)*(y2)-(x2)*(y1) #define DET3(x1,y1,z1,x2,y2,z2,x3,y3,z3) (x1)*(y2)*(z3)+(x2)*(y3)*(z1)+(x3)*(y1)*(z2)-(z1)*(y2)*(x3)-(z2)*(y3)*(x1)-(z3)*(y1)*(x2) #define INC(a) for(auto& v:a)v++; #define DEC(a) for(auto& v:a)v--; #define SQU(x) (x)*(x) #define L0 ll(0) #ifdef ATCODER #include <atcoder/all> using namespace atcoder; using mint = modint1000000007; using mint2 = modint998244353; #endif template<typename T = ll> vector<T> read(size_t n) { vector<T> ts(n); for (size_t i = 0; i < n; i++) cin >> ts[i]; return ts; } template<typename TV, const ll N> void read_tuple_impl(TV&) {} template<typename TV, const ll N, typename Head, typename... Tail> void read_tuple_impl(TV& ts) { get<N>(ts).emplace_back(*(istream_iterator<Head>(cin))); read_tuple_impl<TV, N + 1, Tail...>(ts); } template<typename... Ts> decltype(auto) read_tuple(size_t n) { tuple<vector<Ts>...> ts; for (size_t i = 0; i < n; i++) read_tuple_impl<decltype(ts), 0, Ts...>(ts); return ts; } //using val = tuple<mint2, mint2, ll, ll>; //using func = ll; // //val op(val a, val b) { // auto [a1, a2, a3, a4] = a; // auto [b1, b2, b3, b4] = b; // return make_tuple(a1 + b1, a2 + b2, max(a3, b3), a4 + b4); //} // //val e() { return make_tuple(0, 0, 0, 0); } // //val mp(func f, val a) //{ // if (f < 0) // return a; // auto [a1, a2, a3, a4] = a; // return make_tuple(a2 * f, a2, f, a4); //} //func comp(func f, func g) { // if (f < 0) // return g; // return f; //} ll di[4] = { 1,0,-1,0 }; ll dj[4] = { 0,1,0,-1 }; ll si[4] = { 0,3,3,0 }; ll sj[4] = { 0,0,3,3 }; //ll di[4] = { -1,-1,1,1 }; //ll dj[4] = { -1,1,-1,1 }; ll di8[8] = { 0,-1,-1,-1,0,1,1,1 }; ll dj8[8] = { -1,-1,0,1,1,1,0,-1 }; ll x, n, k; VVL g; VL a; void dfs(ll now, ll parent, ll d) { if (d % 2) x ^= a[now] % (k + 1); for (auto& v : g[now]) { if (v == parent) continue; dfs(v, now, d + 1); } } void solve() { cin >> n >> k; g.clear(); g.resize(n); REP(i, n - 1) { ll u, v; cin >> u >> v; u--; v--; g[u].push_back(v); g[v].push_back(u); } a = read(n); x = 0; dfs(0, -1, 0); if (x) { cout << "K\n"; } else { cout << "P\n"; } return; } int main() { ll t = 1; cin >> t; while (t--) { solve(); } return 0; }