Ideone.com

fork download

copy

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define ll long long int
#define ull unsigned long long int
#define pb push_back
#define ff first
#define INF 1e18
#define nl "\n"
#define sp " "
#define ss second
using namespace std;
using namespace __gnu_pbds;
template <typename PB>
using ordered_set = tree<PB, null_type, less_equal<PB>, rb_tree_tag, tree_order_statistics_node_update>;
const ll mod = 1e9 + 7;
 
int lg2(const int &x) { return 31 - __builtin_clz(x); }
ll bigMod(ll a, ll b)
{
    a %= mod;
    ll res = 1;
    while (b > 0)
    {
        if (b & 1)
            res = (res * a) % mod;
        a = (a * a) % mod;
        b >>= 1;
    }
    return res;
}
inline ll MOD(ll a) { return (a % mod + mod) % mod; }
inline ll modAdd(ll a, ll b) { return MOD(MOD(a) + MOD(b)); }
inline ll modSub(ll a, ll b) { return MOD(MOD(a) - MOD(b)); }
inline ll modMul(ll a, ll b) { return MOD(MOD(a) * MOD(b)); }
inline ll modInv(ll a) { return bigMod(a, mod - 2); }
inline ll modDiv(ll a, ll b)
{
    a = MOD(a);
    b = modInv(b);
    return (a * b) % mod;
}
ll lcm(ll a, ll b)
{
    return (a * (b / __gcd(a, b)));
}
ll fix_mod(ll a, ll b)
{
    return (a % b + b) % b;
}
void fastio()
{
    ios_base::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    //#ifndef ONLINE_JUDGE
    // freopen("input.txt", "r+", stdin);
    // freopen("output.txt", "w+", stdout);
    //#endif
}
const int sz = 2e5 + 10;
int n;
int par[sz];
int points[sz];
int sze[sz];
void make(int i)
{
    par[i] = i;
    sze[i] = 1;
    points[i] = 0;
}
int get(int i)
{
    if (i == par[i])
    {
        return i;
    }
    return par[i] = get(par[i]);
}
void join(int a, int b)
{
    a = get(a);
    b = get(b);
    if (a != b)
    {
        if (sze[par[a]] < sze[par[b]])
            swap(a, b);
 
        sze[par[a]] += sze[par[b]];
        par[b] = par[a];
    }
}
void add(int i, int pt)
{
    i=get(i);
    for(int j=1;j<=n;j++)
    {
        if(i==par[j])
        {
            points[j]+=pt;
        }
    }
}
int main()
{
    fastio();
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        make(i);
    }
    int m;
    cin >> m;
    while (m--)
    {
        string s;
        cin >> s;
        if (s == "add")
        {
            int a, b;
            cin >> a >> b;
            add(a, b);
        }
        else if (s == "join")
        {
            int a, b;
            cin >> a >> b;
            join(a, b);
        }
        else
        {
            int a;
            cin >> a;
            cout << points[a] << nl;
        }
    }
}

I2luY2x1ZGUgPGJpdHMvc3RkYysrLmg+CiNpbmNsdWRlIDxleHQvcGJfZHMvYXNzb2NfY29udGFpbmVyLmhwcD4KI2luY2x1ZGUgPGV4dC9wYl9kcy90cmVlX3BvbGljeS5ocHA+CiNkZWZpbmUgbGwgbG9uZyBsb25nIGludAojZGVmaW5lIHVsbCB1bnNpZ25lZCBsb25nIGxvbmcgaW50CiNkZWZpbmUgcGIgcHVzaF9iYWNrCiNkZWZpbmUgZmYgZmlyc3QKI2RlZmluZSBJTkYgMWUxOAojZGVmaW5lIG5sICJcbiIKI2RlZmluZSBzcCAiICIKI2RlZmluZSBzcyBzZWNvbmQKdXNpbmcgbmFtZXNwYWNlIHN0ZDsKdXNpbmcgbmFtZXNwYWNlIF9fZ251X3BiZHM7CnRlbXBsYXRlIDx0eXBlbmFtZSBQQj4KdXNpbmcgb3JkZXJlZF9zZXQgPSB0cmVlPFBCLCBudWxsX3R5cGUsIGxlc3NfZXF1YWw8UEI+LCByYl90cmVlX3RhZywgdHJlZV9vcmRlcl9zdGF0aXN0aWNzX25vZGVfdXBkYXRlPjsKY29uc3QgbGwgbW9kID0gMWU5ICsgNzsKCmludCBsZzIoY29uc3QgaW50ICZ4KSB7IHJldHVybiAzMSAtIF9fYnVpbHRpbl9jbHooeCk7IH0KbGwgYmlnTW9kKGxsIGEsIGxsIGIpCnsKICAgIGEgJT0gbW9kOwogICAgbGwgcmVzID0gMTsKICAgIHdoaWxlIChiID4gMCkKICAgIHsKICAgICAgICBpZiAoYiAmIDEpCiAgICAgICAgICAgIHJlcyA9IChyZXMgKiBhKSAlIG1vZDsKICAgICAgICBhID0gKGEgKiBhKSAlIG1vZDsKICAgICAgICBiID4+PSAxOwogICAgfQogICAgcmV0dXJuIHJlczsKfQppbmxpbmUgbGwgTU9EKGxsIGEpIHsgcmV0dXJuIChhICUgbW9kICsgbW9kKSAlIG1vZDsgfQppbmxpbmUgbGwgbW9kQWRkKGxsIGEsIGxsIGIpIHsgcmV0dXJuIE1PRChNT0QoYSkgKyBNT0QoYikpOyB9CmlubGluZSBsbCBtb2RTdWIobGwgYSwgbGwgYikgeyByZXR1cm4gTU9EKE1PRChhKSAtIE1PRChiKSk7IH0KaW5saW5lIGxsIG1vZE11bChsbCBhLCBsbCBiKSB7IHJldHVybiBNT0QoTU9EKGEpICogTU9EKGIpKTsgfQppbmxpbmUgbGwgbW9kSW52KGxsIGEpIHsgcmV0dXJuIGJpZ01vZChhLCBtb2QgLSAyKTsgfQppbmxpbmUgbGwgbW9kRGl2KGxsIGEsIGxsIGIpCnsKICAgIGEgPSBNT0QoYSk7CiAgICBiID0gbW9kSW52KGIpOwogICAgcmV0dXJuIChhICogYikgJSBtb2Q7Cn0KbGwgbGNtKGxsIGEsIGxsIGIpCnsKICAgIHJldHVybiAoYSAqIChiIC8gX19nY2QoYSwgYikpKTsKfQpsbCBmaXhfbW9kKGxsIGEsIGxsIGIpCnsKICAgIHJldHVybiAoYSAlIGIgKyBiKSAlIGI7Cn0Kdm9pZCBmYXN0aW8oKQp7CiAgICBpb3NfYmFzZTo6c3luY193aXRoX3N0ZGlvKGZhbHNlKSwgY2luLnRpZSgwKSwgY291dC50aWUoMCk7CiAgICAvLyNpZm5kZWYgT05MSU5FX0pVREdFCiAgICAvLyBmcmVvcGVuKCJpbnB1dC50eHQiLCAicisiLCBzdGRpbik7CiAgICAvLyBmcmVvcGVuKCJvdXRwdXQudHh0IiwgIncrIiwgc3Rkb3V0KTsKICAgIC8vI2VuZGlmCn0KY29uc3QgaW50IHN6ID0gMmU1ICsgMTA7CmludCBuOwppbnQgcGFyW3N6XTsKaW50IHBvaW50c1tzel07CmludCBzemVbc3pdOwp2b2lkIG1ha2UoaW50IGkpCnsKICAgIHBhcltpXSA9IGk7CiAgICBzemVbaV0gPSAxOwogICAgcG9pbnRzW2ldID0gMDsKfQppbnQgZ2V0KGludCBpKQp7CiAgICBpZiAoaSA9PSBwYXJbaV0pCiAgICB7CiAgICAgICAgcmV0dXJuIGk7CiAgICB9CiAgICByZXR1cm4gcGFyW2ldID0gZ2V0KHBhcltpXSk7Cn0Kdm9pZCBqb2luKGludCBhLCBpbnQgYikKewogICAgYSA9IGdldChhKTsKICAgIGIgPSBnZXQoYik7CiAgICBpZiAoYSAhPSBiKQogICAgewogICAgICAgIGlmIChzemVbcGFyW2FdXSA8IHN6ZVtwYXJbYl1dKQogICAgICAgICAgICBzd2FwKGEsIGIpOwogICAgICAgIAogICAgICAgIHN6ZVtwYXJbYV1dICs9IHN6ZVtwYXJbYl1dOwogICAgICAgIHBhcltiXSA9IHBhclthXTsKICAgIH0KfQp2b2lkIGFkZChpbnQgaSwgaW50IHB0KQp7CiAgICBpPWdldChpKTsKICAgIGZvcihpbnQgaj0xO2o8PW47aisrKQogICAgewogICAgICAgIGlmKGk9PXBhcltqXSkKICAgICAgICB7CiAgICAgICAgICAgIHBvaW50c1tqXSs9cHQ7CiAgICAgICAgfQogICAgfQp9CmludCBtYWluKCkKewogICAgZmFzdGlvKCk7CiAgICBjaW4gPj4gbjsKICAgIGZvciAoaW50IGkgPSAxOyBpIDw9IG47IGkrKykKICAgIHsKICAgICAgICBtYWtlKGkpOwogICAgfQogICAgaW50IG07CiAgICBjaW4gPj4gbTsKICAgIHdoaWxlIChtLS0pCiAgICB7CiAgICAgICAgc3RyaW5nIHM7CiAgICAgICAgY2luID4+IHM7CiAgICAgICAgaWYgKHMgPT0gImFkZCIpCiAgICAgICAgewogICAgICAgICAgICBpbnQgYSwgYjsKICAgICAgICAgICAgY2luID4+IGEgPj4gYjsKICAgICAgICAgICAgYWRkKGEsIGIpOwogICAgICAgIH0KICAgICAgICBlbHNlIGlmIChzID09ICJqb2luIikKICAgICAgICB7CiAgICAgICAgICAgIGludCBhLCBiOwogICAgICAgICAgICBjaW4gPj4gYSA+PiBiOwogICAgICAgICAgICBqb2luKGEsIGIpOwogICAgICAgIH0KICAgICAgICBlbHNlCiAgICAgICAgewogICAgICAgICAgICBpbnQgYTsKICAgICAgICAgICAgY2luID4+IGE7CiAgICAgICAgICAgIGNvdXQgPDwgcG9pbnRzW2FdIDw8IG5sOwogICAgICAgIH0KICAgIH0KfQ==

Success #stdin #stdout 0.01s 5348KB

stdin

copy

Standard input is empty

stdout

copy

https://ideone.com/ZJt4a5

language:

C++14 (gcc 8.3)

created:

visibility:

secret

Share or Embed source code

Discover > Sphere Engine API

The brand new service which powers Ideone!

Discover > IDE Widget

Widget for compiling and running the source code in a web browser!

Discover > Sphere Engine API

Discover > IDE Widget

Choose your language