oj.uz

Submission #201661

# Submission time Handle Problem Language Result Execution time Memory
201661 2020-02-11T16:19:39 Z SamAnd Zamjene (COCI16_zamjene) C++17
140 / 140
 5641 ms 181824 KB 
#include <bits/stdc++.h>
using namespace std;
#define m_p make_pair

int ka()
{
    int x = 0;
    bool z = false;
    while (1)
    {
        char y = getchar();
        if ('0' <= y && y <= '9')
        {
            x = (x * 10) + (y - '0');
            z = true;
        }
        else
        {
            if (z)
                return x;
        }
    }
}

const int N = 1000006;
const int P = 1000003;
const int M = 1000000007;
const int P1 = 2000003;
const int M1 = 1000000009;
int ast[N], ast1[N];
void pre()
{
    ast[0] = 1;
    ast1[0] = 1;
    for (int i = 1; i < N; ++i)
    {
        ast[i] = (ast[i - 1] * 1LL * P) % M;
        ast1[i] = (ast1[i - 1] * 1LL * P1) % M1;
    }
}

int n, qqq;
int a[N], b[N];

map<pair<int, int>, int> mp;
long long ans4;

void ave(pair<int, int> x, int q)
{
    if (x.first || x.second)
    {
        ans4 += mp[m_p((M - x.first) % M, (M1 - x.second) % M1)] * 1LL * q;
    }
    mp[x] += q;
}

void han(pair<int, int> x, int q)
{
    if (x.first || x.second)
    {
        ans4 -= mp[m_p((M - x.first) % M, (M1 - x.second) % M1)] * 1LL * q;
    }
    mp[x] -= q;
}

int p[N];
int ha[N], hb[N];
int ha1[N], hb1[N];
int qq[N];

int fi(int x)
{
    if (x == p[x])
        return x;
    return p[x] = fi(p[x]);
}

pair<int, int> u(int x)
{
    return m_p((ha[x] - hb[x] + M) % M, (ha1[x] - hb1[x] + M1) % M1);
}

int main()
{
    //freopen("input.txt", "r", stdin);
    pre();
    n = ka();
    qqq = ka();
    for (int i = 1; i <= n; ++i)
    {
        a[i] = ka();
        b[i] = a[i];
    }
    sort(b + 1, b + n + 1);
    for (int i = 1; i <= n; ++i)
    {
        p[i] = i;
        ha[i] = ast[a[i]];
        hb[i] = ast[b[i]];
        ha1[i] = ast1[a[i]];
        hb1[i] = ast1[b[i]];
        qq[i] = 1;
        ave(u(i), qq[i]);
    }
    while (qqq--)
    {
        int ty;
        ty = ka();
        if (ty == 1)
        {
            int x, y;
            x = ka();
            y = ka();
            if (fi(x) == fi(y))
            {
                swap(a[x], a[y]);
                continue;
            }
            int xx = fi(x);
            int yy = fi(y);
            han(u(xx), qq[xx]);
            han(u(yy), qq[yy]);
            ha[xx] = (ha[xx] - ast[a[x]] + M) % M;
            ha[yy] = (ha[yy] - ast[a[y]] + M) % M;
            ha1[xx] = (ha1[xx] - ast1[a[x]] + M1) % M1;
            ha1[yy] = (ha1[yy] - ast1[a[y]] + M1) % M1;
            swap(a[x], a[y]);
            ha[xx] = (ha[xx] + ast[a[x]]) % M;
            ha[yy] = (ha[yy] + ast[a[y]]) % M;
            ha1[xx] = (ha1[xx] + ast1[a[x]]) % M1;
            ha1[yy] = (ha1[yy] + ast1[a[y]]) % M1;
            ave(u(xx), qq[xx]);
            ave(u(yy), qq[yy]);
        }
        else if (ty == 2)
        {
            int x, y;
            x = ka();
            y = ka();
            if (fi(x) == fi(y))
                continue;
            x = fi(x);
            y = fi(y);
            han(u(x), qq[x]);
            han(u(y), qq[y]);
            p[x] = y;
            ha[y] = (ha[y] + ha[x]) % M;
            hb[y] = (hb[y] + hb[x]) % M;
            ha1[y] = (ha1[y] + ha1[x]) % M1;
            hb1[y] = (hb1[y] + hb1[x]) % M1;
            qq[y] += qq[x];
            ave(u(y), qq[y]);
        }
        else if (ty == 3)
        {
            if (mp[m_p(0, 0)] == n)
                printf("DA\n");
            else
                printf("NE\n");
        }
        else
        {
            printf("%lld\n", ans4);
        }
    }
    return 0;
}

Subtask #1
14.0 / 14.0

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

Subtask #2
14.0 / 14.0

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

Subtask #3
14.0 / 14.0

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

Subtask #4
14.0 / 14.0

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

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 8312 KB Output is correct
2 Correct 15 ms 8184 KB Output is correct
3 Correct 16 ms 8312 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 19 ms 8568 KB Output is correct
2 Correct 19 ms 8568 KB Output is correct
3 Correct 19 ms 8696 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 64 ms 11768 KB Output is correct
2 Correct 126 ms 13432 KB Output is correct
3 Correct 179 ms 16632 KB Output is correct

Subtask #8
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1336 ms 40272 KB Output is correct
2 Correct 4815 ms 131028 KB Output is correct
3 Correct 5641 ms 181824 KB Output is correct

Subtask #9
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 2560 ms 81400 KB Output is correct
2 Correct 4197 ms 102380 KB Output is correct
3 Correct 1917 ms 90256 KB Output is correct

Subtask #10
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 1089 ms 47096 KB Output is correct
2 Correct 3032 ms 86612 KB Output is correct
3 Correct 1924 ms 90164 KB Output is correct