結果
問題 |
No.2340 Triple Tree Query (Easy)
|
ユーザー |
![]() |
提出日時 | 2023-05-27 04:12:30 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,307 bytes |
コンパイル時間 | 2,039 ms |
コンパイル使用メモリ | 178,572 KB |
実行使用メモリ | 24,320 KB |
最終ジャッジ日時 | 2024-12-28 12:00:37 |
合計ジャッジ時間 | 59,579 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 31 TLE * 5 |
ソースコード
#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 C, D; cin >> V >> K >> C >> D; V--; X[V] = (X[V] * C + D) % MOD; for (int w : E[V]){ X[w] = (X[w] * C + D) % MOD; } } if (T == 3){ int V; long long C, D; cin >> V >> C >> D; V--; auto dfs = [&](auto dfs, int v) -> void { X[v] = (X[v] * C + D) % MOD; for (int w : c[v]){ dfs(dfs, w); } }; dfs(dfs, V); } } }