Submission #612089

#	Time	Username	Problem	Language	Result	Execution time	Memory
612089		1bin	The Potion of Great Power (CEOI20_potion)	C++14	17 / 100	245 ms	17952 KiB

potion

#include <bits/stdc++.h>

using namespace std;

#define fi first
#define all(v) v.begin(), v.end()
const int NMAX = 1e5 + 5;
const int sqt = 50;
int n, d, u, q, h[NMAX], a, b, A[2 * NMAX], B[2 * NMAX];
vector<int> ix[NMAX];
vector<vector<pair<int, int>>> v[NMAX];
set<pair<int, int>> l, r;

void init(int N, int D, int H[]){
    n = N; d = D;
    for(int i = 0; i < n; i++) h[i] = H[i];
    return;
}

void chk(int x){
    int sz = ix[x].size();
    if(sz % sqt) return;
    set<pair<int, int>> s;
    if(--sz) for(auto& t : v[x].back()) s.emplace(t);
    for(int i = sz * sqt; i < sz * sqt + sqt; i++){
        int y = (A[ix[x][i]] == x) ? B[ix[x][i]] : A[ix[x][i]];
        if(s.find({h[y], y}) == s.end()) s.insert({h[y], y});
        else s.erase({h[y], y});
    }
    vector<pair<int, int>> tmp;
    for(auto t : s) tmp.emplace_back(t);
    v[x].emplace_back(tmp);
    return;
}

void curseChanges(int U, int A_[], int B_[]){
    u = U;
    for(int i = 0; i < u; i++) A[i] = A_[i], B[i] = B_[i];
    for(int i = 0; i < u; i++){
        a = A[i]; b = B[i];
        ix[a].emplace_back(i);
        ix[b].emplace_back(i);
        chk(a); chk(b);
    }
    return;
}

int question(int x, int y, int V){
    int ans = 1e9, xb, yb, lsz, rsz;
    int xsz = lower_bound(all(ix[x]), V) - ix[x].begin();
    int ysz = lower_bound(all(ix[y]), V) - ix[y].begin();
    xb = xsz / sqt; yb = ysz / sqt;
    if(xb >= 1) for(auto& t : v[x][xb - 1]) l.emplace(t);
    if(yb >= 1) for(auto& t : v[y][yb - 1]) r.emplace(t);
    for(int i = xb * sqt; i < xsz; i++) {
        int t = (A[ix[x][i]] == x) ? B[ix[x][i]] : A[ix[x][i]];
        if(l.find({h[t], t}) == l.end()) l.insert({h[t], t});
        else l.erase({h[t], t});
    }
    for(int i = yb * sqt; i < ysz; i++) {
        int t = (A[ix[y][i]] == y) ? B[ix[y][i]] : A[ix[y][i]];
        if(r.find({h[t], t}) == r.end()) r.insert({h[t], t});
        else r.erase({h[t], t});
    }
    auto xi = l.begin();
    auto yi = r.begin();
    lsz = l.size(); rsz = r.size();
    while(xi != l.end() && yi != r.end()){
        ans = min(ans, abs(xi->fi - yi->fi));
        xi->fi < yi->fi ? xi++ : yi++;
    }
   l.clear(); r.clear();
    return ans;
}

Compilation message (stderr)

potion.cpp: In function 'int question(int, int, int)':
potion.cpp:49:28: warning: variable 'lsz' set but not used [-Wunused-but-set-variable]
   49 |     int ans = 1e9, xb, yb, lsz, rsz;
      |                            ^~~
potion.cpp:49:33: warning: variable 'rsz' set but not used [-Wunused-but-set-variable]
   49 |     int ans = 1e9, xb, yb, lsz, rsz;
      |                                 ^~~

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