Submission #1186802

#	Time	Username	Problem	Language	Result	Execution time	Memory
1186802		user192837	Prize (CEOI22_prize)	C++17	0 / 100	1192 ms	157840 KiB

Main

#include <bits/stdc++.h>
#define ar array
using namespace std;

const int N = 5e5 + 6;
vector <int> g[2][N];
int st[2][20][N], d[2][N], root[2], qd;

void dfs1(int x, int par) {
    d[qd][x] = d[qd][par] + 1, st[qd][0][x] = par;
    for (int i = 1; i < 20; i++)
        st[qd][i][x] = st[qd][i - 1][st[qd][i - 1][x]];
    for (int y : g[qd][x])
        if (y != par)
            dfs1(y, x);
}

int jump(int x, int k) {
    int i = 0;
    while (k) {
        if (k & 1)
            x = st[qd][i][x];
        k >>= 1;
        i++;
    }
    return x;
}

int lca(int a, int b) {
    if (d[qd][a] > d[qd][b])
        swap(a, b);
    b = jump(b, d[qd][b] - d[qd][a]);
    if (a == b)
        return a;
    int i = 19;
    while (i >= 0) {
        int na = st[qd][i][a], nb = st[qd][i][b];
        if (na != nb)
            a = na, b = nb;
        i--;
    }
    return st[qd][0][a];
}

int n, k, q, t;
int checkpiont[2][N], taken[N];

void dfs3(int x, int par) {
    if (k > 0)
        k--, taken[x] = 1;
    for (int y : g[0][x])
        if (y != par)
            dfs3(y, x);
}

vector <ar <int, 2>> qrs;

void dfs2(int x, int par) {
    for (int y : g[0][x]) {
        if (y != par) {
            if (taken[y]) {
                cout << "? " << root[0] << ' ' << y << '\n';
                qrs.push_back({root[0], y});
            }
            dfs2(y, x);
        }
    }
}

int main() {
    cin >> n >> k >> q >> t;
    for (int i = 0; i < 2; i++) {
        for (int j = 1; j <= n; j++) {
            int p;
            cin >> p;
            if (~p)
                g[i][p].emplace_back(j), g[i][j].emplace_back(p);
            else
                root[i] = j;
        }
    }
    // qd = 0, dfs1(root[0], 0);
    // qd = 1, dfs1(root[1], 0);
    dfs3(root[0], 0);
    for (int i = 1; i <= n; i++)
        if (taken[i])
            cout << i << ' ';
    cout << '\n';
    dfs2(root[0], 0);
    cout << "!\n";
    for (int j = 0; j < qrs.size(); j++) {
        int y = qrs[j][1];
        vector <int> res(4);
        for (int &i : res)
            cin >> i;
        qd = 1; int lc = lca(root[0], y);
        checkpiont[0][y] = res[1];
        checkpiont[1][lc] = res[2];
        if (lc == root[0])
            checkpiont[1][y] = res[3];
        else
            checkpiont[1][y] = res[2] + res[3];
    }
    qd = 1, dfs1(root[0], 0);
    while (t--) {
        int a, b;
        cin >> a >> b;
        qd = 0; cout << checkpiont[0][a] + checkpiont[0][b] - 2 * checkpiont[0][lca(a, b)] << ' ';
        qd = 1; cout << checkpiont[1][a] + checkpiont[1][b] - 2 * checkpiont[1][lca(a, b)] << '\n';
    }
    // cout;
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 25
• Subtask 3: 0 / 19
• Subtask 4: 0 / 22
• Subtask 5: 0 / 24