結果
問題 | No.1054 Union add query |
ユーザー |
![]() |
提出日時 | 2020-05-15 21:48:37 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 211 ms / 2,000 ms |
コード長 | 2,396 bytes |
コンパイル時間 | 889 ms |
コンパイル使用メモリ | 76,228 KB |
最終ジャッジ日時 | 2025-01-10 11:31:44 |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 |
ソースコード
#include <iostream> #include <algorithm> #include <utility> #include <vector> #include <numeric> template <class T, class U> inline bool chmin(T &lhs, const U &rhs) { if (lhs > rhs) { lhs = rhs; return true; } return false; } template <class T, class U> inline bool chmax(T &lhs, const U &rhs) { if (lhs < rhs) { lhs = rhs; return true; } return false; } // [l, r) from l to r struct range { struct itr { int i; constexpr itr(int i_): i(i_) { } constexpr void operator ++ () { ++i; } constexpr int operator * () const { return i; } constexpr bool operator != (itr x) const { return i != x.i; } }; const itr l, r; constexpr range(int l_, int r_): l(l_), r(std::max(l_, r_)) { } constexpr itr begin() const { return l; } constexpr itr end() const { return r; } }; // [l, r) from r to l struct revrange { struct itr { int i; constexpr itr(int i_): i(i_) { } constexpr void operator ++ () { --i; } constexpr int operator * () const { return i; } constexpr bool operator != (itr x) const { return i != x.i; } }; const itr l, r; constexpr revrange(int l_, int r_): l(l_ - 1), r(std::max(l_, r_) - 1) { } constexpr itr begin() const { return r; } constexpr itr end() const { return l; } }; int main() { std::ios::sync_with_stdio(false); std::cin.tie(nullptr); int N, Q; std::cin >> N >> Q; std::vector<int> parent(N, -1); std::vector<int> weight(N); auto find = [&](int x) { while (parent[x] >= 0) { x = parent[x]; } return x; }; auto add = [&](int x, int v) { weight[find(x)] += v; }; auto unite = [&](int x, int y) { x = find(x); y = find(y); if (x == y) { return; } if (parent[x] > parent[y]) { std::swap(x, y); } parent[x] += parent[y]; parent[y] = x; weight[y] -= weight[x]; }; auto fold = [&](int x) { int res = weight[x]; while (parent[x] >= 0) { x = parent[x]; res += weight[x]; } return res; }; while (Q--) { int t; std::cin >> t; if (t == 1) { int a, b; std::cin >> a >> b; --a; --b; unite(a, b); } else if (t == 2) { int a, b; std::cin >> a >> b; --a; add(a, b); } else { int a, b; std::cin >> a >> b; --a; --b; std::cout << fold(a) << '\n'; } } return 0; }