Submission #726251

# Submission time Handle Problem Language Result Execution time Memory
726251 2023-04-18T17:12:16 Z bitaro1337 Prize (CEOI22_prize) C++14
0 / 100
2803 ms 991880 KB
//BITARO BITARO BITARO
#include <bits/stdc++.h>
using namespace std;

using ll = long long;

const int K = 20;
struct tree {
    int n, root, tt = 0;
    vector<ll> dist;
    vector<vector<pair<int, ll>>> adj;
    vector<vector<int>> jump, g;
    vector<bool> vis;
    vector<int> time_in, depth;
    tree(int _n, int _root) {
        n = _n, root = _root;
        adj.resize(n);
        g.resize(n);
    }
    void init() {
        dist.assign(n, 0);
        depth.assign(n, 0);
        vis.assign(n, false);
        jump.assign(n, vector<int>(K, root));
        time_in.assign(n, 0);
        dfs(root, -1);
        for(int j = 1; j < K; ++j) {
            for(int i = 0; i < n; ++i) {
                jump[i][j] = jump[jump[i][j - 1]][j - 1];
            }
        }
    }
    void add_edge(int a, int b) {
        if(a >= n || b >= n) exit(0);
        g[a].push_back(b);
    }
    void add_edge(int a, int b, ll w) {
        if(a >= n || b >= n) exit(0);
        adj[a].push_back({b, w});
    }
    void dfs(int u, int par) {
        time_in[u] = tt++;
        for(int v: g[u]) {
            if(v == par) continue;
            depth[v] = depth[u] + 1;
            dfs(v, u);
        }
    }
    void get_distances(int u) {
        vis[u] = true;
        for(auto x: adj[u]) {
            int v = x.first;
            ll w = x.second;
            if(vis[v]) continue;
            dist[v] = dist[u] + w;
            get_distances(v);
        }
    }
    int lca(int a, int b) {
        if(depth[a] < depth[b]) swap(a, b);
        for(int j = K - 1; j >= 0; --j) {
            if(depth[jump[a][j]] >= depth[b]) a = jump[a][j];
        }
        if(a == b) return a;
        for(int j = K - 1; j >= 0; --j) {
            if(jump[a][j] != jump[b][j]) a = jump[a][j], b = jump[b][j];
        }
        return jump[a][0];
    }
    int get_depth(int u) {return dist[u];}
    int tm(int x) {return time_in[x];}

};
int time1[(int)1e6 + 5], time2[(int)1e6 + 5];
bool cmp1(int x, int y) {
    return time1[x] < time1[y];
}
bool cmp2(int x, int y) {
    return time2[x] < time2[y];
}
int main() {
    int n, k, q, t; cin >> n >> k >> q >> t;
    vector<int> p1(n), p2(n);
    int root1 = -1, root2 = -1;
    for(int i = 0; i < n; ++i) {
        cin >> p1[i]; --p1[i];
        if(p1[i] == -2) root1 = i;
    }
    for(int i = 0; i < n; ++i) {
        cin >> p2[i]; --p2[i];
        if(p2[i] == -2) root2 = i;
    }
    if(root1 == -1 || root2 == -1) exit(0);
    tree t1(n, root1), t2(n, root2);
    for(int i = 0; i < n; ++i) {
        if(p1[i] != -2) t1.add_edge(p1[i], i);
        if(p2[i] != -2) t2.add_edge(p2[i], i);
    }
    t1.init(); t2.init();
    vector<int> nodes(n); iota(nodes.begin(), nodes.end(), 0);
    for(int i = 0; i < n; ++i) {
        time1[i] = t1.tm(i);
        time2[i] = t2.tm(i);
    }
    sort(nodes.begin(), nodes.end(), cmp1);
    while((int)nodes.size() > k) nodes.pop_back();
    if(t1.tm(root2) >= k) {
        nodes.pop_back();
        nodes.push_back(root2);
    }
    for(int x: nodes) cout << x + 1 << " ";
    cout << "\n"; cout.flush();
    sort(nodes.begin(), nodes.end(), cmp2);
    for(int i = 0; i < k - 1; ++i) {
        cout << "? " << nodes[i] + 1 << " " << nodes[i + 1] + 1 << "\n";
    }
    cout << "!\n"; cout.flush();
    for(int i = 0; i < k - 1; ++i) {
        {
            int u = nodes[i], v = nodes[i + 1];
            int l = t1.lca(u, v);
            int c1, c2; cin >> c1 >> c2;
            t1.add_edge(l, u, c1);
            t1.add_edge(l, v, c2);
            t1.add_edge(u, l, -c1);
            t1.add_edge(v, l, -c2);
        }
        {
            int u = nodes[i], v = nodes[i + 1];
            int l = t2.lca(u, v);
            int c1, c2; cin >> c1 >> c2;
            t2.add_edge(l, u, c1);
            t2.add_edge(l, v, c2);
            t2.add_edge(u, l, -c1);
            t2.add_edge(v, l, -c2);
        }
    }
    t1.get_distances(root1);
    t2.get_distances(root2);
    vector<pair<int, int>> queries(t);
    for(int i = 0; i < t; ++i) {
        cin >> queries[i].first >> queries[i].second; --queries[i].first, --queries[i].second;
    }
    assert(0);
    for(auto x: queries) {
        {
            int l = t1.lca(x.first, x.second);
            cout << t1.get_depth(x.first) + t1.get_depth(x.second) - 2 * t1.get_depth(l) << " ";
        }
        {
            int l = t2.lca(x.first, x.second);
            cout << t2.get_depth(x.first) + t2.get_depth(x.second) - 2 * t2.get_depth(l) << "\n";
        }
    }
    cout.flush();
}
# Verdict Execution time Memory Grader output
1 Runtime error 1429 ms 503144 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 1527 ms 510636 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 1203 ms 481596 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 2553 ms 963052 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Runtime error 2803 ms 991880 KB Execution killed with signal 6
2 Halted 0 ms 0 KB -