oj.uz

답안 #139896

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
139896 2019-08-01T15:45:28 Z stefdasca Zamjene (COCI16_zamjene) C++14
126 / 140
 6000 ms 192476 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;


ll mul(ll a, ll b)
{
    return (a*b) % mod;
}
int n, q;
int v[NMAX];
int st[NMAX];
int tt[NMAX], sz[NMAX];
ll hsh[NMAX], sthsh[NMAX], pw[NMAX];

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

# 결과 실행 시간 메모리 Grader output
1 Correct 9 ms 8156 KB Output is correct
2 Correct 9 ms 8184 KB Output is correct
3 Correct 10 ms 8312 KB Output is correct

Subtask #2
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 9 ms 8184 KB Output is correct
2 Correct 9 ms 8156 KB Output is correct
3 Correct 9 ms 8316 KB Output is correct

Subtask #3
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 9 ms 8184 KB Output is correct
2 Correct 10 ms 8220 KB Output is correct
3 Correct 10 ms 8312 KB Output is correct

Subtask #4
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 10 ms 8284 KB Output is correct
2 Correct 15 ms 8272 KB Output is correct
3 Correct 10 ms 8216 KB Output is correct

Subtask #5
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 10 ms 8184 KB Output is correct
2 Correct 12 ms 8312 KB Output is correct
3 Correct 10 ms 8312 KB Output is correct

Subtask #6
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 15 ms 8668 KB Output is correct
2 Correct 15 ms 8696 KB Output is correct
3 Correct 15 ms 8696 KB Output is correct

Subtask #7
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 81 ms 11740 KB Output is correct
2 Correct 141 ms 13348 KB Output is correct
3 Correct 239 ms 16696 KB Output is correct

Subtask #8
0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1455 ms 40028 KB Output is correct
2 Correct 5521 ms 130864 KB Output is correct
3 Execution timed out 6101 ms 192476 KB Time limit exceeded

Subtask #9
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2872 ms 81224 KB Output is correct
2 Correct 5729 ms 114748 KB Output is correct
3 Correct 2056 ms 103416 KB Output is correct

Subtask #10
14.0 / 14.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1199 ms 47176 KB Output is correct
2 Correct 3450 ms 86200 KB Output is correct
3 Correct 2069 ms 103532 KB Output is correct