oj.uz

Submission #83184

# Submission time Handle Problem Language Result Execution time Memory
83184 2018-11-05T22:47:24 Z fredbr Pictionary (COCI18_pictionary) C++17
140 / 140
 209 ms 25248 KB 
#include <bits/stdc++.h>

using namespace std;

using ii = pair<int, int>;

template <int maxn>
struct Dsu {
    int pai[maxn], w[maxn];

    int find(int x)
    {
        if (x == pai[x]) return x;
        return pai[x] = find(pai[x]);
    }

    bool check(int a, int b)
    {
        a = find(a), b = find(b);
        return a == b;
    }
    void join(int a, int b)
    {
        a = find(a), b = find(b);
        
        if (a == b) return;
        if (w[a] < w[b]) swap(a, b);

        pai[b] = a;
        w[a] += w[b];
    }

    void init(int n) {
        fill(w+1, w+n+1, 1);
        iota(pai+1, pai+n+1, 1);
    }
};

struct Query
{
    int ff, ss, id;
};

int const maxn = 101010;

int n, m, q, lk;
Dsu<maxn> dsu;
vector<Query> qr[maxn], nqr[maxn];
vector<ii> qe, nq;
int res[maxn];

void clear()
{
    swap(qe, nq), swap(qr, nqr), lk = m+1, dsu.init(n);
    nq.clear();
}

void add(int k)
{
    while (lk > k) {
        --lk;
        for (int j = 2*lk; j <= n; j += lk)
            dsu.join(lk, j);
    }
}

void split(int l, int k, int r)
{
    bool isl = false, isr = false;
    for (auto p : qr[l]) {
        if (!dsu.check(p.ff, p.ss))
            isl = true, nqr[l].push_back(p);
        else
            isr = true, nqr[k].push_back(p);
    }
    if (isr) nq.push_back({k, r});
    if (isl) nq.push_back({l, k});
    qr[l].clear();
}

void add_ans(int l)
{
    for (auto p : qr[l])
        res[p.id] = l;
}

int main()
{
    ios::sync_with_stdio(false), cin.tie(0);

    cin >> n >> m >> q;

    dsu.init(n);
    qe.push_back({1, m+1});

    for (int i = 0; i < q; i++) {
        int a, b;
        cin >> a >> b;

        qr[1].push_back({a, b, i});
    }

    lk = m+1;
    for (int i = 0; i < qe.size(); i++) {

        int l = qe[i].first, r = qe[i].second;
        
        add((l+r)/2);      

        if (l+1 < r) split(l, (l+r)/2, r);    
        else add_ans(l);

        if (i == int(qe.size())-1) clear(), i = -1;
    }

    for (int i = 0; i < q; i++) cout << m-res[i]+1 << "\n";
}

Compilation message

pictionary.cpp: In function 'int main()':
pictionary.cpp:104:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for (int i = 0; i < qe.size(); i++) {
                     ~~^~~~~~~~~~~

Subtask #1
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 12 ms 5500 KB Output is correct

Subtask #2
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 6648 KB Output is correct

Subtask #3
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 46 ms 10184 KB Output is correct
2 Correct 39 ms 10184 KB Output is correct

Subtask #4
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 55 ms 12944 KB Output is correct
2 Correct 49 ms 12944 KB Output is correct

Subtask #5
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 61 ms 12944 KB Output is correct
2 Correct 61 ms 12944 KB Output is correct

Subtask #6
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 73 ms 13984 KB Output is correct
2 Correct 71 ms 13984 KB Output is correct

Subtask #7
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 110 ms 17044 KB Output is correct
2 Correct 91 ms 17044 KB Output is correct

Subtask #8
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 70 ms 17044 KB Output is correct
2 Correct 143 ms 19632 KB Output is correct

Subtask #9
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 161 ms 22352 KB Output is correct
2 Correct 172 ms 22352 KB Output is correct

Subtask #10
14.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 209 ms 25248 KB Output is correct
2 Correct 188 ms 25248 KB Output is correct