Submission #713354

#	Time	Username	Problem	Language	Result	Execution time	Memory
713354		Stickfish	Cop and Robber (BOI14_coprobber)	C++17	16 / 100	83 ms	5696 KiB

coprobber

#include "coprobber.h"
#include <iostream>
#include <cassert>
#include <bitset>
#include <vector>
using namespace std;

bitset<MAX_N> edg[MAX_N];
bitset<MAX_N> redg[MAX_N];
int bad_edg_cnt[2][MAX_N][MAX_N];
int depth[2][MAX_N][MAX_N];
int n;
int C;
int pR = -1;

int start(int N, bool A[MAX_N][MAX_N]) {
    n = N;
    int deg[MAX_N];

    for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) {
        edg[i][j] = redg[j][i] = A[i][j];
    }
    for (int i = 0; i < N; ++i)
        deg[i] = edg[i].count();
    for (int i = 0; i < N; ++i) for (int j = 0; j < N; ++j) {
        bad_edg_cnt[0][i][j] = 1;
        bad_edg_cnt[1][i][j] = deg[j];
        depth[1][i][j] = depth[0][i][j] = 1234567890;
    }
    vector<vector<int>> q;
    for (int i = 0; i < N; ++i) {
        bad_edg_cnt[0][i][i] = bad_edg_cnt[1][i][i] = 0;
        q.push_back({0, i, i});
        q.push_back({1, i, i});
    }
    int timer = 0;
    while (q.size()) {
        int t = q.back()[0];
        int u = q.back()[1];
        int v = q.back()[2];
        depth[t][u][v] = timer++;
        //cout << t << ' ' << u << ' ' << v << endl;
        q.pop_back();
        bitset<MAX_N> cedg;
        if (t == 0)
            cedg = redg[v];
        else {
            cedg = redg[u];
            //cedg[u] = 1;
        }
        for (int i = cedg._Find_first(); i < n; i = cedg._Find_next(i)) {
            int nu = u;
            int nv = v;
            if (t == 0) 
                nv = i;
            else
                nu = i;
            if (nu == nv)
                continue;
            if (--bad_edg_cnt[t ^ 1][nu][nv] == 0) {
                q.push_back({t ^ 1, nu, nv});
            }
        }
    }
    for (int u = 0; u < n; ++u) for (int v = 0; v < n; ++v) {
        if (u == v || bad_edg_cnt[1][u][v] > 0)
            continue;
        for (int nv = edg[v]._Find_first(); nv < MAX_N; nv = edg[v]._Find_next(nv)) {
            assert(bad_edg_cnt[0][u][nv] <= 0 && depth[0][u][nv] < depth[1][u][v]);
        }
    }
    for (int u = 0; u < n; ++u) {
        bool isok = true;
        for (int v = 0; v < n; ++v) {
            isok &= bad_edg_cnt[0][u][v] <= 0;
        }
        if (isok)
            return C = u;
    }
    return -1; 
}

int nextMove(int R) {
    //if (pR != -1)
        //cout << "A " << C << ' ' << pR << ' ' << depth[1][C][pR] << endl;
    //cout << "B " << C << ' ' << R << ' ' << depth[0][C][R] << endl;
    if (pR != -1) {
        assert(edg[pR][R]);
        assert(bad_edg_cnt[1][C][pR] <= 0);
        assert(depth[1][C][pR] > depth[0][C][R]);
    }
    //assert(bad_edg_cnt[0][C][R] <= 0);
    pR = R;
    for (int nC = edg[C]._Find_first(); nC < n; nC = edg[C]._Find_next(nC)) { 
        if (bad_edg_cnt[1][nC][R] <= 0 && depth[1][nC][R] < depth[0][C][R]) {
            //cout << "C " << nC << ' ' << R << ' ' << depth[1][nC][R] << endl;
            return C = nC;
        }
    }
    //return -1;
    assert(0);
}

• Subtask 1: 16 / 16
• Subtask 2: 0 / 14
• Subtask 3: 0 / 30
• Subtask 4: 0 / 40