Submission #1155435

#TimeUsernameProblemLanguageResultExecution timeMemory
1155435NonozeCop and Robber (BOI14_coprobber)C++20
0 / 100
52 ms15688 KiB
/*
*	Author: Nonoze
*	Created: Saturday 22/02/2025
*/
#include <bits/stdc++.h>
#include "coprobber.h"
using namespace std;

#ifndef IN_LOCAL
	#define dbg(...)
#endif

// #define cout cerr << "OUT: "
#define endl '\n'
#define endlfl '\n' << flush
#define quit(x) return (void)(cout << x << endl)

template<typename T> void read(T& x) { cin >> x;}
template<typename T1, typename T2> void read(pair<T1, T2>& p) { read(p.first), read(p.second);}
template<typename T> void read(vector<T>& v) { for (auto& x : v) read(x); }
template<typename T1, typename T2> void read(T1& x, T2& y) { read(x), read(y); }
template<typename T1, typename T2, typename T3> void read(T1& x, T2& y, T3& z) { read(x), read(y), read(z); }
template<typename T1, typename T2, typename T3, typename T4> void read(T1& x, T2& y, T3& z, T4& zz) { read(x), read(y), read(z), read(zz); }
template<typename T> void print(vector<T>& v) { for (auto& x : v) cout << x << ' '; cout << endl; }

#define sz(x) (int)(x.size())
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
#define make_unique(v) sort(all(v)), v.erase(unique(all(v)), (v).end())
#define pb push_back
#define mp(a, b) make_pair(a, b)
#define fi first
#define se second
#define cmin(a, b) a = min(a, b)
#define cmax(a, b) a = max(a, b)
#define YES cout << "YES" << endl
#define NO cout << "NO" << endl
#define QYES quit("YES")
#define QNO quit("NO")

// #define int long long
#define double long double
const int inf = numeric_limits<int>::max() / 4;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
const int MOD = 1e9+7, LOG=20;

int n;
vector<vector<int>> adj;
vector<vector<vector<pair<int, int>>>> memo;

pair<int, int> dp(int a, int b, bool turn) {
    if (a == b) return {turn, a};
    if (memo[a][b][turn].fi != -1) return memo[a][b][turn];
    if (turn) memo[a][b][turn]={0, -1};
    else memo[a][b][turn]={1, a};
    pair<int, int> ans = {0, -1};
    if (turn) {
        for (auto v: adj[a]) {
            auto [x, y] = dp(v, b, !turn);
            if (!x) {
                ans={1, v};
                break;
            }
        }
        if (ans==mp(0, -1)) {
            auto [x, y]=dp(a, b, !turn);
            if (!x) ans={1, a};
        }
    } else {
        for (auto v: adj[b]) {
            auto [x, y] = dp(a, v, !turn);
            if (!x) {
                ans={1, a};
                break;
            }
        }
    }
    return memo[a][b][turn]=ans;
}

int X;

int start(int N, bool A[MAX_N][MAX_N]) {
    n = N;
    adj.resize(n);
    for (int i=0; i<n; i++) {
        for (int j=0; j<n; j++) {
            if (A[i][j]) {
                adj[i].pb(j);
            }
        }
    }
    memo.resize(n, vector<vector<pair<int, int>>>(n, vector<pair<int, int>>(2, {-1, -1})));
    int ans=-1;
    for (int i=0; i<n; i++) {
        bool ok=1;
        for (int j=0; j<n; j++) {
            if (!dp(i, j, 1).fi) {
                ok=0;
            }
        }
        if (ok) {
            ans=i;
            break;
        }
    }
    X=ans;
    return ans;
}

int nextMove(int R) {
    int nxt=dp(X, R, 1).se;
    X=nxt;
    return nxt;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...