Submission #1204352

#	Time	Username	Problem	Language	Result	Execution time	Memory
1204352		perekopskad	Sphinx's Riddle (IOI24_sphinx)	C++20	28.50 / 100	22 ms	1436 KiB

sphinx

#include "sphinx.h"
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

#define fr(i, l, r) for(ll (i) = l; (i) <= r; (i)++)
#define frb(i, r, l) for(ll (i) = r; (i) >= l; (i)--)
#define pb push_back
#define el '\n'

ll const N = 500 + 10;

vector <ll> g[N];
bool used[N];

void dfs(ll v) {
    used[v] = 1;
    for(ll u : g[v]) 
        if(!used[u])
            dfs(u);
}

std::vector<int> find_colours(int N, std::vector<int> X, std::vector<int> Y) {
    fr(i, 0, X.size() - 1) {
        ll u = X[i], v = Y[i];
        g[u].pb(v);
        g[v].pb(u);
    }

    vector <int> ans(N);

    vector <vector <int>> vec;
    vec.pb({0});

    fr(i, 1, N - 1) {
        vector <int> c(N, N);

        fr(j, 0, i)
            c[j] = -1;
        
        ll cnt = vec.size() + (i < N - 1);
        ll res = perform_experiment(c);

        if(res != cnt) {
            vec.pb({int(i)});
            continue;
        }

        ll l = 0, r = vec.size() - 1;
        while(l < r) {
            ll mid = (l + r) / 2;
            // check [l; mid]
            c.assign(N, N);

            fr(j, l, mid)
                for(ll x : vec[j])
                    c[x] = -1;

            c[i] = -1;

            ll cnt = mid - l + 2;
            ll res = perform_experiment(c);

            if(cnt == res) r = mid;
            else l = mid + 1;
        }
        vec[l].pb(i);
    }

    fr(i, 0, vec.size() - 1)
        for(ll x : vec[i])
            ans[x] = i;
    
    return ans;
}

• Subtask 0: 0 / 0
• Subtask 1: 1.5 / 3
• Subtask 2: 0 / 7
• Subtask 3: 16.5 / 33
• Subtask 4: 10.5 / 21
• Subtask 5: 0 / 36