제출 #963616

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
963616		Ghetto	The Potion of Great Power (CEOI20_potion)	C++17	56 / 100	3023 ms	46852 KiB

potion

#include <bits/stdc++.h>
using namespace std;
using pii = pair<int, int>;
const int MAX_N = 1e5 + 5, K = 50, MAX_T = 2e5 + 5;
 
int n;
int h[MAX_N]; // Not id
int t;
int e1[MAX_T], e2[MAX_T]; // Not id
 
int id[MAX_N], rev_id[MAX_N]; 
vector<pii> upds[MAX_N]; // {time, node} and {0, -1}
set<int> adj[MAX_N]; // reused
vector<vector<int>> crit_adjs[MAX_N];
void update1(int u, int v, set<int>& s) { // one-way
    if (s.count(v)) s.erase(v);
    else s.insert(v);
}
void save(int u) {
    vector<int> cur_adj(adj[u].begin(), adj[u].end());
    crit_adjs[u].push_back(cur_adj);
}
void precomp() {
    vector<pii> order;
    for (int i = 0; i < n; i++) order.push_back({h[i], i});
    sort(order.begin(), order.end());
    for (int i = 0; i < n; i++) {
        id[order[i].second] = i;
        rev_id[i] = order[i].second;
    }
 
    for (int upd = 0; upd <= t; upd++) {
        if (upd == 0) {
            for (int u = 0; u < n; u++) {
                upds[u].push_back({0, -1});
                save(u);
            }
            continue;
        }
 
        int u = id[e1[upd]], v = id[e2[upd]];
        update1(u, v, adj[u]);
        update1(v, u, adj[v]);
        upds[u].push_back({upd, v});
        upds[v].push_back({upd, u});
 
        if (((int) upds[u].size() - 1) % K == 0) save(u);
        if (((int) upds[v].size() - 1) % K == 0) save(v);
    }
}
 
void init(int tmp_n, int tmp_d, int tmp_h[]) {
    n = tmp_n;
    for (int i = 0; i < n; i++) h[i] = tmp_h[i];
}
void curseChanges(int tmp_t, int tmp_e1[], int tmp_e2[]) {
    t = tmp_t;
    for (int i = 1; i <= t; i++) {
        e1[i] = tmp_e1[i - 1];
        e2[i] = tmp_e2[i - 1];
    }
    precomp();
}
 
int question(int u, int v, int upd) {
    u = id[u]; 
    v = id[v];

    //

    int hi = upper_bound(upds[u].begin(), upds[u].end(), make_pair(upd, (int) 1e9)) - upds[u].begin() - 1; 
    int lo = K * (hi / K);
    set<int> u_adj(crit_adjs[u][hi / K].begin(), crit_adjs[u][hi / K].end());
    for (int i = lo + 1; i <= hi; i++) 
        update1(u, upds[u][i].second, u_adj);

    int hi2 = upper_bound(upds[v].begin(), upds[v].end(), make_pair(upd, (int) 1e9)) - upds[v].begin() - 1; 
    int lo2 = K * (hi2 / K);
    set<int> v_adj(crit_adjs[v][hi2 / K].begin(), crit_adjs[v][hi2 / K].end());

    for (int i = lo2 + 1; i <= hi2; i++) 
        update1(v, upds[v][i].second, v_adj);

    //

    int ans = 1e9;
    auto l_it = v_adj.begin(), r_it = v_adj.begin();
    for (auto it = u_adj.begin(); it != u_adj.end(); it++) {
        while (l_it != v_adj.end() && next(l_it, 1) != v_adj.end() && h[rev_id[*next(l_it, 1)]] <= h[rev_id[*it]]) l_it++;
        while (r_it != v_adj.end() && h[rev_id[*r_it]] <= h[rev_id[*it]]) r_it++;
 
        if (l_it != v_adj.end()) ans = min(ans, abs(h[rev_id[*it]] - h[rev_id[*l_it]]));
        if (r_it != v_adj.end()) ans = min(ans, abs(h[rev_id[*it]] - h[rev_id[*r_it]]));
    }    
    return ans;
}

• Subtask 1: 0 / 0
• Subtask 2: 17 / 17
• Subtask 3: 0 / 14
• Subtask 4: 18 / 18
• Subtask 5: 21 / 21
• Subtask 6: 0 / 30