結果
問題 |
No.2341 Triple Tree Query (Medium)
|
ユーザー |
![]() |
提出日時 | 2023-05-25 23:28:01 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,649 bytes |
コンパイル時間 | 2,086 ms |
コンパイル使用メモリ | 178,852 KB |
実行使用メモリ | 18,176 KB |
最終ジャッジ日時 | 2024-12-28 12:13:40 |
合計ジャッジ時間 | 17,834 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | WA * 2 |
other | RE * 36 |
ソースコード
#include <bits/stdc++.h> using namespace std; const long long MOD = 998244353; int main(){ int N, Q; cin >> N >> Q; vector<vector<int>> E(N); for (int i = 0; i < N - 1; i++){ int A, B; cin >> A >> B; A--; B--; E[A].push_back(B); E[B].push_back(A); } vector<long long> X(N); for (int i = 0; i < N; i++){ cin >> X[i]; } vector<int> p(N, -1); vector<vector<int>> c(N); vector<int> d(N, 0); queue<int> q; q.push(0); while (!q.empty()){ int v = q.front(); q.pop(); for (int w : E[v]){ if (w != p[v]){ p[w] = v; c[v].push_back(w); d[w] = d[v] + 1; q.push(w); } } } for (int i = 0; i < Q; i++){ int T; cin >> T; if (T == 1){ int V; cin >> V; V--; cout << X[V] << endl; } if (T == 2){ int V, K; long long B, C; cin >> V >> K >> B >> C; V--; X[V] = (X[V] * B + C) % MOD; for (int w : E[V]){ X[w] = (X[w] * B + C) % MOD; } } if (T == 3){ int U, V; long long B, C; cin >> U >> V >> B >> C; U--; V--; while (true){ if (U == V){ X[U] = (X[U] * B + C) % MOD; break; } if (d[U] > d[V]){ swap(U, V); } X[V] = (X[V] * B + C) % MOD; V = p[V]; } } if (T == 4){ int V; long long B, C; cin >> V >> B >> C; V--; auto dfs = [&](auto dfs, int v) -> void { X[v] = (X[v] * B + C) % MOD; for (int w : c[v]){ dfs(dfs, w); } }; dfs(dfs, V); } } }