제출 #83184

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
83184		fredbr	Pictionary (COCI18_pictionary)	C++17	140 / 140	209 ms	25248 KiB

pictionary

#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";
}

컴파일 시 표준 에러 (stderr) 메시지

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 / 14
• Subtask 2: 14 / 14
• Subtask 3: 14 / 14
• Subtask 4: 14 / 14
• Subtask 5: 14 / 14
• Subtask 6: 14 / 14
• Subtask 7: 14 / 14
• Subtask 8: 14 / 14
• Subtask 9: 14 / 14
• Subtask 10: 14 / 14