oj.uz

답안 #595277

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
595277 2022-07-13T14:13:39 Z MadokaMagicaFan 늑대인간 (IOI18_werewolf) C++17
100 / 100
 881 ms 83588 KB 
#include "bits/stdc++.h"

using namespace std;

using ll = long long;
using vi = vector<int>;

#define all(v)                      v.begin(), v.end()
#define sz(v)                       ((int)v.size())

#define pb                          push_back
#define endl                        '\n'

#define fst                         first
#define scn                         second

const int N = 2e5;
const int K = 18;
vi adj[N];

struct dsu{
    int s[N];
    int in[N];
    vector<array<int,K>> p;
    vector<vi> dch;
    int n;

    dsu(int _n) {
        n = _n;
        p.resize(n);
        dch.resize(n);
        for (int i = 0; i < n; ++i) {
            p[i][0] = i;
            p[i][1] = i;
            s[i] = 1;
        }
    }

    int
    getp(int x)
    {
        return p[x][1] = (x == p[x][1]) ? x : getp(p[x][1]) ;
    }

    void
    conc(int a, int b)
    {
        b = getp(b);
        if (a == b) return;

        dch[a].pb(b);
        p[b][0] = a;
        p[b][1] = a;
        s[a] += s[b];
        return;
    }

    int
    dfs(int x, int cnt)
    {
        in[x] = cnt++;
        for(int u : dch[x])
            cnt = dfs(u, cnt);
        return cnt;
    }

    void
    gen()
    {
        dfs(n-1,0);
        for (int j = 1; j < K; ++j) {
            for (int i = 0; i < n; ++i) {
                p[i][j] = p[p[i][j-1]][j-1];
            }
        }
        return;
    }

    void
    query(int& x1, int& x2, int st, int v)
    {
        ++v;
        for (int j = K-1; j >= 0; --j) {
            if (p[st][j] < v) st = p[st][j];
        }

        x1 = in[st];
        x2 = x1 + s[st]-1;
        return;
    }
};

struct aib{
    vi tr;
    int n;

    aib(int _n) {
        n = _n;
        tr.assign(n,0);
    }

    void
    add(int x, int v)
    {
        for(; x < n; x |= (x+1))
            tr[x] += v;
        return;
    }

    int
    query(int x)
    {
        int res = 0;
        for (; x >= 0; x = (x&(x+1))-1)
            res += tr[x];
        return res;
    }

    int
    query(int l, int r)
    {
        return query(r) - query(l-1);
    }
};


vector<int> 
check_validity(int n, vector<int> x, vector<int> y, 
        vector<int> b, vector<int> e, vector<int> l, vector<int> r) {
    for (int i = 0; i < sz(x); ++i) {
        adj[x[i]].pb(y[i]);
        adj[y[i]].pb(x[i]);
    }

    dsu up(n), dn(n);


    for (int i = 0; i < n; ++i) {
        for (int u : adj[i])
            if (u < i) up.conc(i, u);
    }

    for (int i = n-1; i >= 0; --i) {
        for (int u : adj[i])
            if (u > i) dn.conc(n-1-i, n-1-u);
    }

    up.gen();
    dn.gen();

    int q = sz(l);
    vi ans(q,0);
    deque<pair<int,int>> in, out;


    vi y1(q) ,y2(q), pos(n);

    for (int i = 0; i < q; ++i) {
        int x1, x2, x3, x4;

        up.query(x1,x2, e[i], r[i]);
        dn.query(x3,x4, n-1-b[i], n-1-l[i]);

        in.pb({x1,i});
        out.pb({x2,i});

        y1[i] = x3;
        y2[i] = x4;
    }

    sort(in.begin(), in.end());
    sort(out.begin(), out.end());

    for (int i = 0; i < n; ++i) pos[up.in[i]] = n-i-1;

    int pnt;

    aib tr(n);

    for (int i = 0; i < n; ++i) {
        while(sz(in)) {
            if (i < in.front().fst) break;

            pnt = in.front().scn;
            in.pop_front();
            ans[pnt] -= tr.query(y1[pnt], y2[pnt]);
        }

        tr.add(dn.in[pos[i]],1);

        while(sz(out)) {
            if (i < out.front().fst) break;

            pnt = out.front().scn;
            out.pop_front();
            ans[pnt] += tr.query(y1[pnt], y2[pnt]);
        }
    }

    for (int i = 0; i < q; ++i)
        if (ans[i]) ans[i] = 1;


    return ans;
}

#ifdef ONPC
void
solve()
{
    int n, m, q;
    cin >> n >> m >> q;

    vector<int> x(m), y(m), s(q), e(q), l(q), r(q);
    for (int i = 0; i < m; ++i) cin >> x[i];
    for (int i = 0; i < m; ++i) cin >> y[i];
    for (int i = 0; i < q; ++i) cin >> s[i];
    for (int i = 0; i < q; ++i) cin >> e[i];
    for (int i = 0; i < q; ++i) cin >> l[i];
    for (int i = 0; i < q; ++i) cin >> r[i];

    vector<int> ans = check_validity(n,x,y,s,e,l,r);

    for (int a : ans) {
        cout << a << ' ';
    }
    cout << endl;
}

int32_t
main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);
    freopen("in", "r", stdin);
    int t = 1;
    /* cin >> t; */
    while(t--)
        solve();
}
#endif

Subtask #1
7.0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Correct 4 ms 8148 KB Output is correct
2 Correct 4 ms 8148 KB Output is correct
3 Correct 4 ms 8020 KB Output is correct
4 Correct 4 ms 8108 KB Output is correct
5 Correct 4 ms 8148 KB Output is correct
6 Correct 4 ms 8148 KB Output is correct
7 Correct 4 ms 8148 KB Output is correct
8 Correct 4 ms 8148 KB Output is correct
9 Correct 4 ms 8148 KB Output is correct

Subtask #2
8.0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Correct 4 ms 8148 KB Output is correct
2 Correct 4 ms 8148 KB Output is correct
3 Correct 4 ms 8020 KB Output is correct
4 Correct 4 ms 8108 KB Output is correct
5 Correct 4 ms 8148 KB Output is correct
6 Correct 4 ms 8148 KB Output is correct
7 Correct 4 ms 8148 KB Output is correct
8 Correct 4 ms 8148 KB Output is correct
9 Correct 4 ms 8148 KB Output is correct
10 Correct 9 ms 9172 KB Output is correct
11 Correct 9 ms 9172 KB Output is correct
12 Correct 8 ms 9080 KB Output is correct
13 Correct 10 ms 9172 KB Output is correct
14 Correct 9 ms 9164 KB Output is correct
15 Correct 10 ms 9172 KB Output is correct

Subtask #3
34.0 / 34.0

# 결과 실행 시간 메모리 Grader output
1 Correct 619 ms 76948 KB Output is correct
2 Correct 618 ms 78412 KB Output is correct
3 Correct 597 ms 77428 KB Output is correct
4 Correct 568 ms 77104 KB Output is correct
5 Correct 599 ms 77040 KB Output is correct
6 Correct 603 ms 76972 KB Output is correct
7 Correct 685 ms 76892 KB Output is correct
8 Correct 687 ms 78316 KB Output is correct
9 Correct 609 ms 77420 KB Output is correct
10 Correct 552 ms 77120 KB Output is correct
11 Correct 526 ms 77020 KB Output is correct
12 Correct 568 ms 76932 KB Output is correct
13 Correct 652 ms 83040 KB Output is correct
14 Correct 693 ms 83036 KB Output is correct
15 Correct 631 ms 82920 KB Output is correct
16 Correct 687 ms 82952 KB Output is correct
17 Correct 610 ms 76940 KB Output is correct

Subtask #4
51.0 / 51.0

# 결과 실행 시간 메모리 Grader output
1 Correct 4 ms 8148 KB Output is correct
2 Correct 4 ms 8148 KB Output is correct
3 Correct 4 ms 8020 KB Output is correct
4 Correct 4 ms 8108 KB Output is correct
5 Correct 4 ms 8148 KB Output is correct
6 Correct 4 ms 8148 KB Output is correct
7 Correct 4 ms 8148 KB Output is correct
8 Correct 4 ms 8148 KB Output is correct
9 Correct 4 ms 8148 KB Output is correct
10 Correct 9 ms 9172 KB Output is correct
11 Correct 9 ms 9172 KB Output is correct
12 Correct 8 ms 9080 KB Output is correct
13 Correct 10 ms 9172 KB Output is correct
14 Correct 9 ms 9164 KB Output is correct
15 Correct 10 ms 9172 KB Output is correct
16 Correct 619 ms 76948 KB Output is correct
17 Correct 618 ms 78412 KB Output is correct
18 Correct 597 ms 77428 KB Output is correct
19 Correct 568 ms 77104 KB Output is correct
20 Correct 599 ms 77040 KB Output is correct
21 Correct 603 ms 76972 KB Output is correct
22 Correct 685 ms 76892 KB Output is correct
23 Correct 687 ms 78316 KB Output is correct
24 Correct 609 ms 77420 KB Output is correct
25 Correct 552 ms 77120 KB Output is correct
26 Correct 526 ms 77020 KB Output is correct
27 Correct 568 ms 76932 KB Output is correct
28 Correct 652 ms 83040 KB Output is correct
29 Correct 693 ms 83036 KB Output is correct
30 Correct 631 ms 82920 KB Output is correct
31 Correct 687 ms 82952 KB Output is correct
32 Correct 610 ms 76940 KB Output is correct
33 Correct 754 ms 76940 KB Output is correct
34 Correct 355 ms 32380 KB Output is correct
35 Correct 820 ms 78312 KB Output is correct
36 Correct 719 ms 77284 KB Output is correct
37 Correct 705 ms 77868 KB Output is correct
38 Correct 682 ms 77592 KB Output is correct
39 Correct 647 ms 82160 KB Output is correct
40 Correct 881 ms 83588 KB Output is correct
41 Correct 623 ms 77436 KB Output is correct
42 Correct 587 ms 77268 KB Output is correct
43 Correct 762 ms 81268 KB Output is correct
44 Correct 631 ms 77908 KB Output is correct
45 Correct 596 ms 82388 KB Output is correct
46 Correct 628 ms 82144 KB Output is correct
47 Correct 621 ms 83112 KB Output is correct
48 Correct 696 ms 83152 KB Output is correct
49 Correct 639 ms 83108 KB Output is correct
50 Correct 670 ms 83020 KB Output is correct
51 Correct 810 ms 83276 KB Output is correct
52 Correct 842 ms 83328 KB Output is correct