답안 #133815

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
133815 2019-07-21T13:16:32 Z osaaateiasavtnl Cats or Dogs (JOI18_catdog) C++14
컴파일 오류
0 ms 0 KB
#include<bits/stdc++.h>
using namespace std;
#define ii pair <int, int>
#define app push_back
mt19937 rnd(228);
struct Node {
    int x, y, cnt;
    Node *l, *r;
    Node (int x_) {
        x = x_; y = rnd();
        l = r = NULL;
        cnt = 1;
    }   
};  
int cnt(Node *t) {
    if (!t) return 0;
    else return t->cnt;
}   
void relax(Node *t) {
    t->cnt = cnt(t->l) + cnt(t->r) + 1;
}   
Node *merge(Node *l, Node *r) {
    if (!l) return r;
    if (!r) return l;
    if (l->y < r->y) {
        l->r = merge(l->r, r);
        relax(l);
        return l;
    }  
    else {
        r->l = merge(l, r->l);
        relax(r);
        return r;
    }   
}   
pair <Node*, Node*> split(Node *t, int x) {
    if (!t) return {NULL, NULL};
    if (t->x < x) {
        auto tmp = split(t->r, x);
        t->r = tmp.first;
        relax(t);
        return {t, tmp.second};
    }
    else {
        auto tmp = split(t->l, x);
        t->l = tmp.second;
        relax(t);
        return {tmp.first, t};
    }   
}
void insert(Node *&t, int x) {
    auto tmp = split(t, x);
    t = merge(merge(tmp.first, new Node(x)), tmp.second);
}   
void erase(Node *&t, int x) {
    auto tmp = split(t, x);
    auto tmp1 = split(tmp.second, x + 1);
    t = merge(tmp.first, tmp1.second);
}   
Node *get(Node *&t, int l, int r) {
    auto tmp = split(t, l);
    auto tmp1 = split(tmp.second, r + 1);
    t = merge(tmp.first, tmp1.second);
    return tmp1.first;
}   
const int N = 1e5 + 7;
Node *dd[N][2];
vector <int> g[N];
int vert[N];
int l[N], r[N], ptr = 0;
void dfs(int u, int p) {
    l[u] = ++ptr;
    vert[ptr] = u;
    for (int v : g[u]) {
        if (v != p) {
            dfs(v, u);
        }   
    }   
    r[u] = ptr;
}
int mx[N << 2];
void upd(int v, int tl, int tr, int p, int x) {
    if (tl == tr) {
        mx[v] = x;
        return;
    }   
    int tm = (tl + tr) >> 1;
    if (p <= tm) upd(v * 2 + 1, tl, tm, p, x);
    else upd(v * 2 + 2, tm + 1, tr, p, x);
    mx[v] = max(mx[v * 2 + 1], mx[v * 2 + 2]);
}   
int parent(int v, int tl, int tr, int l, int r, int x) {
    if (tr < l || r < tl || mx[v] < x) return -1;
    if (tl == tr) return vert[tl];
    int tm = (tl + tr) >> 1;
    int t = parent(v * 2 + 2, tm + 1, tr, l, r, x);
    if (t != -1) return t;
    else return parent(v * 2 + 1, tl, tm, l, r, x);
}   
int ans = 0;
void initialize(int n, int *a, int *b) {
    for (int i = 0; i < n - 1; ++i) {
        g[a[i]].app(b[i]);
        g[b[i]].app(a[i]);
    }   
    ++n;
    g[0].app(1);
    dfs(0, 0);
    for (int i = 0; i < (N << 2); ++i) {
        mx[i] = -1;
    }   
    upd(0, 0, N, l[0], r[0]);
}   
//0 - cat, 1 - dog
int type[N];
void add(int u) {
    int p = parent(0, 0, N, 0, l[u] - 1, r[u]);
    ans += type[u] ^ type[p];
    for (int i = 0; i < 2; ++i) {
        dd[u][i] = get(dd[p][i], l[u], r[u]);        
        if (type[u] ^ type[p]) {
            if (i == type[u]) {
                ans -= cnt(dd[u][i]);
            }   
            else {
                ans += cnt(dd[u][i]);
            }   
        } 
    }   
    insert(dd[p][type[u]], l[u]);
}   
void del(int u) {
    int p = parent(0, 0, N, 0, l[u] - 1, r[u]);
    ans -= type[u] ^ type[p];
    erase(dd[p][type[u]], l[u]);
    for (int i = 0; i < 2; ++i) {
        if (type[u] ^ type[p]) {
            if (i == type[u]) {
                ans += cnt(dd[u][i]);
            }   
            else {
                ans -= cnt(dd[u][i]);
            }   
        }   
        auto tmp = split(dd[p][i], l[u]);
        auto tmp1 = split(dd[p][i], r[u] + 1);
        dd[p][i] = merge(merge(tmp.first, dd[u][i]), tmp1.second);
        dd[u][i] = NULL;
    }   
}   
int f() {
    if (cnt(dd[0][0]) < cnt(dd[0][1])) return ans;
    else return ans - cnt(dd[0][0]) + cnt(dd[0][1]);    
}   
int cat(int u) {
    --u;
    type[u] = 0;
    add(u);
    return f();
}
int dog(int u) {
    --u;
    type[u] = 1;
    add(u);    
    return f();
}
int neighbor(int u) {
    --u;
    del(u);
    return f();        
}   

Compilation message

/tmp/ccIqARp8.o: In function `main':
grader.cpp:(.text.startup+0x157): undefined reference to `initialize(int, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >)'
collect2: error: ld returned 1 exit status