답안 #636736

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
636736 2022-08-30T01:10:19 Z Cross_Ratio Long Mansion (JOI17_long_mansion) C++14
100 / 100
334 ms 107204 KB
#include <bits/stdc++.h>
using namespace std;
const int INF = 1e9;
int N;
int C[500005];
vector<vector<int>> A;
struct SegTree {
    vector<int> seg;
    int MAX;
    SegTree(int N) {
        int i;
        for(i=1;i<2*N;i*=2) {}
        seg.resize(i);
        MAX = i;
        for(i=0;i<MAX;i++) {
            seg[i] = INF;
        }
    }
    void cons() {
        for(int i = MAX/2-1;i>=1;i--) {
            seg[i] = min(seg[2*i], seg[2*i+1]);
        }
    }
    void update(int n) {
        n += MAX/2;
        n /= 2;
        while(n) {
            seg[n] = min(seg[2*n], seg[2*n+1]);
            n /= 2;
        }
    }
    void Edit(int n, int a) {
        seg[n+MAX/2] = a;
        update(n);
    }
    int sum(int s, int e, int n, int ns, int ne) {
        if(e<=ns||ne<=s) return INF;
        if(s<=ns&&ne<=e) return seg[n];
        int mid = ns + ne >> 1;
        return min(sum(s, e, 2*n, ns, mid), sum(s, e, 2*n+1, mid, ne));
    }
    int get_right(int s, int e, int k, int n, int ns, int ne) {
        if(e<=ns||ne<=s||seg[n]>=k) return -1;
        if(n>=MAX/2) return ns;
        int mid = (ns + ne) / 2;
        int c = get_right(s, e, k, 2*n, ns, mid);
        if(c != -1) return c;
        return get_right(s, e, k, 2*n+1, mid, ne);
    }
};
int L[500005];
int R[500005];
int prv[500005];
int nxt[500005];
int D[500005];
void init(vector<int> _C, vector<vector<int>> _A) {
    N = _A.size();
    A = _A;
    int i, j;
    for(i=0;i<N-1;i++) C[i] = _C[i] - 1;
    for(i=0;i<N;i++) {
        for(j=0;j<A[i].size();j++) A[i][j]--;
    }
    for(i=0;i<N;i++) D[i] = -1;
    for(i=0;i<N;i++) {
        prv[i] = (i == 0 ? -1 : D[C[i-1]]);
        for(int n : A[i]) D[n] = i;
    }
    for(i=0;i<N;i++) D[i] = N;
    for(i=N-1;i>=0;i--) {
        nxt[i] = (i == N-1 ? N : D[C[i]]);
        for(int n : A[i]) D[n] = i;
    }
    SegTree tree(N+5);
    int MAX = tree.MAX;
    for(i=0;i<N;i++) {
        tree.seg[i+MAX/2] = prv[i];
        //tree.Edit(i, prv[i]);
    }
    tree.cons();
    for(i=0;i<N;i++) {
        L[i] = i;
        R[i] = i;
        int pL = L[i], pR = R[i];
        while(true) {
            if(R[i] < N-1) R[i] = tree.get_right(R[i]+1, N, L[i], 1, 0, MAX/2) - 1;
            if(R[i]==-2) R[i] = N-1;
            if(L[i] >= 1 && nxt[L[i]-1] <= R[i]) {
                L[i] = L[L[i]-1];
            }
            if(pL==L[i]&&pR==R[i]) break;
            pL= L[i];
            pR = R[i];
        }
        //cout << i << " : " << L[i] <<' ' << R[i] << '\n';
    }
}
bool isPos(int a, int b) {
    a--;
    b--;
    return L[a]<=b&&b<=R[a];
}






signed main() {
    cin.sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int N;
    cin >> N;
    int i, j;
    vector<int> c(N-1);
    for(i=0;i<N-1;i++) cin >> c[i];
    vector<vector<int>> a(N);
    for(i=0;i<N;i++) {
        int sz;
        cin >> sz;
        a[i].resize(sz);
        for(j=0;j<sz;j++) cin >> a[i][j];
    }
    init(c, a);
    int Q;
    cin >> Q;
    while(Q--) {
        int x, y;
        cin >> x >> y;
        cout << (isPos(x, y) ? "YES\n": "NO\n");
    }
}

Compilation message

long_mansion.cpp: In member function 'int SegTree::sum(int, int, int, int, int)':
long_mansion.cpp:39:22: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   39 |         int mid = ns + ne >> 1;
      |                   ~~~^~~~
long_mansion.cpp: In function 'void init(std::vector<int>, std::vector<std::vector<int> >)':
long_mansion.cpp:62:18: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   62 |         for(j=0;j<A[i].size();j++) A[i][j]--;
      |                 ~^~~~~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 724 KB Output is correct
2 Correct 3 ms 948 KB Output is correct
3 Correct 4 ms 1620 KB Output is correct
4 Correct 3 ms 852 KB Output is correct
5 Correct 2 ms 732 KB Output is correct
6 Correct 3 ms 852 KB Output is correct
7 Correct 2 ms 852 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 724 KB Output is correct
2 Correct 3 ms 948 KB Output is correct
3 Correct 4 ms 1620 KB Output is correct
4 Correct 3 ms 852 KB Output is correct
5 Correct 2 ms 732 KB Output is correct
6 Correct 3 ms 852 KB Output is correct
7 Correct 2 ms 852 KB Output is correct
8 Correct 89 ms 6636 KB Output is correct
9 Correct 88 ms 6592 KB Output is correct
10 Correct 89 ms 7080 KB Output is correct
11 Correct 86 ms 7780 KB Output is correct
12 Correct 76 ms 6008 KB Output is correct
13 Correct 100 ms 6856 KB Output is correct
14 Correct 83 ms 6836 KB Output is correct
15 Correct 82 ms 6860 KB Output is correct
16 Correct 101 ms 6612 KB Output is correct
17 Correct 79 ms 6764 KB Output is correct
18 Correct 80 ms 6824 KB Output is correct
19 Correct 105 ms 6888 KB Output is correct
20 Correct 84 ms 6732 KB Output is correct
21 Correct 82 ms 6712 KB Output is correct
22 Correct 94 ms 6732 KB Output is correct
23 Correct 84 ms 6604 KB Output is correct
24 Correct 84 ms 6680 KB Output is correct
25 Correct 92 ms 6636 KB Output is correct
26 Correct 81 ms 6536 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 167 ms 21436 KB Output is correct
2 Correct 178 ms 26284 KB Output is correct
3 Correct 155 ms 26008 KB Output is correct
4 Correct 166 ms 26508 KB Output is correct
5 Correct 167 ms 26440 KB Output is correct
6 Correct 145 ms 25424 KB Output is correct
7 Correct 150 ms 25256 KB Output is correct
8 Correct 169 ms 25236 KB Output is correct
9 Correct 138 ms 25192 KB Output is correct
10 Correct 148 ms 25132 KB Output is correct
11 Correct 174 ms 25116 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 3 ms 724 KB Output is correct
2 Correct 3 ms 948 KB Output is correct
3 Correct 4 ms 1620 KB Output is correct
4 Correct 3 ms 852 KB Output is correct
5 Correct 2 ms 732 KB Output is correct
6 Correct 3 ms 852 KB Output is correct
7 Correct 2 ms 852 KB Output is correct
8 Correct 89 ms 6636 KB Output is correct
9 Correct 88 ms 6592 KB Output is correct
10 Correct 89 ms 7080 KB Output is correct
11 Correct 86 ms 7780 KB Output is correct
12 Correct 76 ms 6008 KB Output is correct
13 Correct 100 ms 6856 KB Output is correct
14 Correct 83 ms 6836 KB Output is correct
15 Correct 82 ms 6860 KB Output is correct
16 Correct 101 ms 6612 KB Output is correct
17 Correct 79 ms 6764 KB Output is correct
18 Correct 80 ms 6824 KB Output is correct
19 Correct 105 ms 6888 KB Output is correct
20 Correct 84 ms 6732 KB Output is correct
21 Correct 82 ms 6712 KB Output is correct
22 Correct 94 ms 6732 KB Output is correct
23 Correct 84 ms 6604 KB Output is correct
24 Correct 84 ms 6680 KB Output is correct
25 Correct 92 ms 6636 KB Output is correct
26 Correct 81 ms 6536 KB Output is correct
27 Correct 167 ms 21436 KB Output is correct
28 Correct 178 ms 26284 KB Output is correct
29 Correct 155 ms 26008 KB Output is correct
30 Correct 166 ms 26508 KB Output is correct
31 Correct 167 ms 26440 KB Output is correct
32 Correct 145 ms 25424 KB Output is correct
33 Correct 150 ms 25256 KB Output is correct
34 Correct 169 ms 25236 KB Output is correct
35 Correct 138 ms 25192 KB Output is correct
36 Correct 148 ms 25132 KB Output is correct
37 Correct 174 ms 25116 KB Output is correct
38 Correct 311 ms 87960 KB Output is correct
39 Correct 324 ms 107204 KB Output is correct
40 Correct 279 ms 67364 KB Output is correct
41 Correct 334 ms 105572 KB Output is correct
42 Correct 158 ms 26384 KB Output is correct
43 Correct 151 ms 26488 KB Output is correct
44 Correct 216 ms 45724 KB Output is correct
45 Correct 232 ms 45948 KB Output is correct
46 Correct 231 ms 46100 KB Output is correct
47 Correct 162 ms 26436 KB Output is correct
48 Correct 148 ms 26220 KB Output is correct
49 Correct 244 ms 45756 KB Output is correct
50 Correct 217 ms 45872 KB Output is correct
51 Correct 231 ms 46228 KB Output is correct
52 Correct 231 ms 44760 KB Output is correct
53 Correct 274 ms 68000 KB Output is correct
54 Correct 322 ms 88992 KB Output is correct
55 Correct 263 ms 68164 KB Output is correct