제출 #959669

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
959669		vjudge1	게임 (IOI14_game)	C++17	15 / 100	2 ms	512 KiB

game

#pragma GCC optimize("Ofast,unroll-loops,no-stack-protector,fast-math,inline")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,lzcnt,mmx,abm,avx,avx2,fma")
#include "game.h"
#include <bits/stdc++.h>
using namespace std;

const int N = 1500;

int n, x, y, curr, need, cnt, p[N], sz[N];
unordered_set<int> vis, adj[N];

int findSet(int x) {
    return (p[x] == x ? x : p[x] = findSet(p[x]));
}

void unite(int x, int y) {
    x = findSet(x);
    y = findSet(y);

    if (sz[x] < sz[y]) swap(x, y);

    p[y] = x;
    sz[x] += sz[y];

    while (!adj[y].empty()) {
        int z = *adj[y].begin();
        
        adj[z].insert(x);
        adj[x].insert(z);

        adj[z].erase(y);
        adj[y].erase(adj[y].begin());
    }
}

void addEdge(int x, int y) {
    adj[x].insert(y);
    adj[y].insert(x);

    unite(x, y);
}

void removeEdge(int x, int y) {
    adj[x].erase(y);
    adj[y].erase(x);
}

void dfs(int x) {
    vis.insert(x);

    cnt += sz[x];
    if (cnt >= need) return;

    for (int y : adj[x]) {
        if (vis.find(y) != vis.end()) continue;
        dfs(y);
    }
}

void initialize(int NN) {
    n = NN;

    curr = 0;
    need = 1;
    for (int i = 0; i < n; i++) {
        p[i] = i;
        sz[i] = 1;

        for (int j = i+1; j < n; j++) {
            adj[i].insert(j);
            adj[j].insert(i);
        }
    }
}

int hasEdge(int u, int v) {
    curr++;
    if (curr >= (n-need)*need) need++;

    u = findSet(u);
    v = findSet(v);
    if (u == v) return 0;

    removeEdge(u, v);

    vis.clear();
    cnt = 0;
    dfs(u);
    if (cnt < need) {
        addEdge(u, v);
        return 1;
    }

    vis.clear();
    cnt = 0;
    dfs(v);
    if (cnt < need) {
        addEdge(u, v);
        return 1;
    }

    return 0;
}

• Subtask 1: 15 / 15
• Subtask 2: 0 / 27
• Subtask 3: 0 / 58