Submission #579600

#	Time	Username	Problem	Language	Result	Execution time	Memory
579600		wiwiho	Game (APIO22_game)	C++17	2 / 100	33 ms	720 KiB

game

#include "game.h"

#include <bits/stdc++.h>
#include <bits/extc++.h>

#define StarBurstStream ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define iter(a) a.begin(), a.end()
#define riter(a) a.rbegin(), a.rend()
#define lsort(a) sort(iter(a))
#define gsort(a) sort(riter(a))
#define pb(a) push_back(a)
#define eb(a) emplace_back(a)
#define pf(a) push_front(a)
#define ef(a) emplace_front(a)
#define pob pop_back()
#define pof pop_front()
#define mp(a, b) make_pair(a, b)
#define F first
#define S second
#define mt make_tuple
#define gt(t, i) get<i>(t)
#define tomax(a, b) ((a) = max((a), (b)))
#define tomin(a, b) ((a) = min((a), (b)))
#define topos(a) ((a) = (((a) % MOD + MOD) % MOD))
#define uni(a) a.resize(unique(iter(a)) - a.begin())
#define printv(a, b) {bool pvaspace=false; \
for(auto pva : a){ \
    if(pvaspace) b << " "; pvaspace=true;\
    b << pva;\
}\
b << "\n";}

using namespace std;
using namespace __gnu_pbds;

typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;

using pii = pair<int, int>;
using pll = pair<ll, ll>;
using pdd = pair<ld, ld>;
using tiii = tuple<int, int, int>;

const ll MOD = 1000000007;
const ll MAX = 2147483647;

template<typename A, typename B>
ostream& operator<<(ostream& o, pair<A, B> p){
    return o << '(' << p.F << ',' << p.S << ')';
}

ll ifloor(ll a, ll b){
    if(b < 0) a *= -1, b *= -1;
    if(a < 0) return (a - b + 1) / b;
    else return a / b;
}

ll iceil(ll a, ll b){
    if(b < 0) a *= -1, b *= -1;
    if(a > 0) return (a + b - 1) / b;
    else return a / b;
}

int n, k;
vector<set<int>> ls, rs;
vector<deque<int>> g, rg;
vector<vector<int>> tg;
deque<pii> todo;

bool check(){
    for(int i = 0; i < k; i++){
        vector<bool> vst(k + 1);
        queue<int> q;
        q.push(i);
        while(!q.empty()){
            int now = q.front();
            q.pop();
            for(int v : tg[now]){
                if(vst[v]) continue;
                q.push(v);
                vst[v] = true;
            }
        }
        if(vst[i]) return true;
    }
    return false;
}

void build(int L = 0, int R = k - 1){
    if(L > R) return;
    int M = (L + R) / 2;
    for(int i = L; i < M; i++) ls[M].insert(i);
    for(int i = M + 1; i <= R; i++) rs[M].insert(i);
    build(L, M - 1);
    build(M + 1, R);
}

int ans = 0;
bool ok = false;

void dfs(int now){
    while(!g[now].empty()){
        int v = g[now].front();
        g[now].pof;
        todo.eb(mp(now, v));
        dfs(v);
    }
}

void dfsr(int now){
    while(!rg[now].empty()){
        int v = rg[now].front();
        rg[now].pof;
        todo.eb(mp(v, now));
        dfsr(v);
    }
}

void addls(int M, int id){
    if(rs[M].find(id) != rs[M].end() || M == id){
        ans = 1;
    }
    ls[M].insert(id);
    dfsr(id);
}

void addrs(int M, int id){
    if(ls[M].find(id) != ls[M].end() || M == id){
        ans = 1;
    }
    rs[M].insert(id);
    dfs(id);
}

void add(int u, int v, int L = 0, int R = k - 1){
    if(L > R) return;
    int M = (L + R) / 2;
    if(v == M || ls[M].find(v) != ls[M].end()){
        //cerr << "left\n";
        addls(M, u);
        add(u, v, L, M - 1);
        ok = true;
    }
    if(u == M || rs[M].find(u) != rs[M].end()){
        //cerr << "right\n";
        addrs(M, v);
        add(u, v, M + 1, R);
        ok = true;
    }
    /*cerr << "add " << mp(u, v) << " " << M << " " << L << " " << R << "\n";
    printv(ls[M], cerr);
    printv(rs[M], cerr);*/
}

void init(int _n, int _k){
    n = _n;
    k = _k;
    ls.resize(n);
    rs.resize(n);
    g.resize(n);
    rg.resize(n);
    tg.resize(n);
    for(int i = 0; i < k - 1; i++){
        tg[i].eb(i + 1);
    }
    build();
}

int add_teleporter(int u, int v){
    tg[u].eb(v);
    ok = false;
    todo.eb(mp(u, v));
    while(!todo.empty()){
        pii e = todo.front();
        todo.pof;
        //cerr << "owo " << e << "\n";
        add(e.F, e.S);
        if(ans == 1) break;
    }
    //cerr << "ok " << ok << "\n";
    if(!ok){
        g[u].eb(v);
        rg[v].eb(u);
    }
    assert(ans == check());
    return ans;
}

• Subtask 1: 2 / 2
• Subtask 2: 0 / 10
• Subtask 3: 0 / 18
• Subtask 4: 0 / 30
• Subtask 5: 0 / 40