結果
問題 | No.2295 Union Path Query (Medium) |
ユーザー |
![]() |
提出日時 | 2023-05-05 22:52:18 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 568 ms / 4,000 ms |
コード長 | 1,601 bytes |
コンパイル時間 | 1,731 ms |
コンパイル使用メモリ | 170,164 KB |
実行使用メモリ | 6,528 KB |
最終ジャッジ日時 | 2024-11-23 10:44:13 |
合計ジャッジ時間 | 28,683 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 53 |
ソースコード
#include <bits/stdc++.h> using namespace std; const int INF = 1000000001; const long long MOD = 998244353; struct unionfind{ vector<int> p, t; unionfind(int N): p(N, -1), t(N, INF){ } int root(int x){ if (p[x] < 0){ return x; } else { return root(p[x]); } } bool same(int x, int y){ return root(x) == root(y); } int size(int v){ return -p[root(v)]; } void unite(int x, int y, int w){ x = root(x); y = root(y); if (p[x] < p[y]){ swap(x, y); } p[y] += p[x]; p[x] = y; t[x] = w; } }; int main(){ int N, X, Q; cin >> N >> X >> Q; unionfind UF(N); vector<long long> ans(N, 0); for (int i = 0; i < Q; i++){ int t; cin >> t; if (t == 1){ int v, w; cin >> v >> w; int u = X; u = UF.root(u); v = UF.root(v); if (u != v){ long long res = ans[u] + ans[v] + (long long) UF.size(u) * UF.size(v) % MOD * w; res %= MOD; UF.unite(u, v, w); ans[UF.root(u)] = res; } } if (t == 2){ int u, v; cin >> u >> v; if (!UF.same(u, v)){ cout << -1 << endl; } else { int ans = 0; while (u != v){ if (UF.t[u] > UF.t[v]){ swap(u, v); } ans = UF.t[u]; u = UF.p[u]; } cout << ans << endl; X += ans; X %= N; } } if (t == 3){ int v; cin >> v; cout << ans[UF.root(v)] << endl; } if (t == 4){ int value; cin >> value; X += value; X %= N; } } }