oj.uz

Submission #139898

# Submission time Handle Problem Language Result Execution time Memory
139898 2019-08-01T15:46:56 Z stefdasca Zamjene (COCI16_zamjene) C++14
140 / 140
 2426 ms 133072 KB 
#include<bits/stdc++.h>
#pragma GCC optimize("O3")
#define fi first
#define se second
#define pb push_back
#define pf push_front
#define mod 1000000007
#define NMAX 1000010

using namespace std;

typedef long long ll;

int n, q;
int v[NMAX];
int st[NMAX];
int tt[NMAX], sz[NMAX];
ll hsh[NMAX], sthsh[NMAX], pw[NMAX];

unordered_map<ll, int> diff_nodes;
ll tot_pairs;

void add_nodes(int sgn, ll diff, int m)
{
    if(diff != 0)
        tot_pairs += sgn * m * diff_nodes[-diff];
    diff_nodes[diff] += sgn * m;
}
int Find(int a)
{
    if(tt[a] == a)
        return a;
    return tt[a] = Find(tt[a]);
}
void Union(int a, int b)
{
    add_nodes(-1, hsh[a] - sthsh[a], sz[a]);
    add_nodes(-1, hsh[b] - sthsh[b], sz[b]);
    if(sz[a] > sz[b])
    {
        tt[b] = a, sz[a] += sz[b];
        hsh[a] = hsh[a]+hsh[b];
        sthsh[a] = sthsh[a]+sthsh[b];
        add_nodes(1, hsh[a] - sthsh[a], sz[a]);
    }
    else
    {
        tt[a] = b, sz[b] += sz[a];
        hsh[b] = hsh[a]+hsh[b];
        sthsh[b] = sthsh[a]+sthsh[b];
        add_nodes(1, hsh[b] - sthsh[b], sz[b]);
    }

}
void add(int sgn, int u, int pos, int value)
{
  sthsh[u] += sgn * pw[pos];
  hsh[u] += sgn * pw[value];
  sz[u] += sgn;
}
void swp(int a, int b)
{
    int u = Find(a);
    int vv = Find(b);
    add_nodes(-1, hsh[u] - sthsh[u], sz[u]);
    add_nodes(-1, hsh[vv] - sthsh[vv], sz[vv]);
    add(-1, Find(a), a, v[a]);
    add(-1, Find(b), b, v[b]);

    swap(v[a], v[b]);

    add(1, Find(a), a, v[a]);
    add(1, Find(b), b, v[b]);
    add_nodes(1, hsh[u] - sthsh[u], sz[u]);
    add_nodes(1, hsh[vv] - sthsh[vv], sz[vv]);
}
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n >> q;
    pw[0] = 1;
    for(int i = 1; i < NMAX; ++i)
        pw[i] = pw[i-1] * 1234577;
    for(int i = 1; i <= n; ++i)
    {
        cin >> v[i];
        st[i] = v[i];
        tt[i] = i;
        sz[i] = 1;
    }
    sort(st + 1, st + n + 1);
    for(int i = 1; i <= n; ++i)
    {
        hsh[i] = pw[v[i]];
        sthsh[i] = pw[st[i]];
        add_nodes(1, hsh[i] - sthsh[i], sz[i]);
    }
    for(int i = 1; i <= q; ++i)
    {
        int tip;
        cin >> tip;
        if(tip == 1)
        {
            int a, b;
            cin >> a >> b;
            if(Find(a) == Find(b))
            {
                swap(v[a], v[b]);
                continue;
            }
            swp(a, b);
        }
        if(tip == 2)
        {
            int a, b;
            cin >> a >> b;
            if(Find(a) != Find(b))
                Union(Find(a), Find(b));
        }
        if(tip == 3)
        {
            if(diff_nodes[0] == n)
                cout << "DA\n";
            else
                cout << "NE\n";
        }
        if(tip == 4)
        {
            cout << tot_pairs << '\n';
        }
    }
    return 0;
}

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 8184 KB Output is correct
2 Correct 11 ms 8184 KB Output is correct
3 Correct 10 ms 8184 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 8188 KB Output is correct
2 Correct 10 ms 8184 KB Output is correct
3 Correct 9 ms 8184 KB Output is correct

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 8156 KB Output is correct
2 Correct 10 ms 8184 KB Output is correct
3 Correct 10 ms 8184 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 8184 KB Output is correct
2 Correct 10 ms 8288 KB Output is correct
3 Correct 10 ms 8276 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 10 ms 8284 KB Output is correct
2 Correct 10 ms 8212 KB Output is correct
3 Correct 10 ms 8312 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 8696 KB Output is correct
2 Correct 14 ms 8696 KB Output is correct
3 Correct 15 ms 8760 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 74 ms 12364 KB Output is correct
2 Correct 91 ms 13340 KB Output is correct
3 Correct 109 ms 15112 KB Output is correct

Subtask #8
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 857 ms 35688 KB Output is correct
2 Correct 1905 ms 92924 KB Output is correct
3 Correct 2426 ms 133072 KB Output is correct

Subtask #9
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1251 ms 69464 KB Output is correct
2 Correct 1868 ms 92164 KB Output is correct
3 Correct 1220 ms 73580 KB Output is correct

Subtask #10
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 850 ms 46432 KB Output is correct
2 Correct 1291 ms 71548 KB Output is correct
3 Correct 1253 ms 73364 KB Output is correct