Submission #1191383

#TimeUsernameProblemLanguageResultExecution timeMemory
1191383onbertSimurgh (IOI17_simurgh)C++17
0 / 100
408 ms3888 KiB
#include "simurgh.h"
#include <bits/stdc++.h>
using namespace std;

const int maxn = 505;
int n;
int id[maxn][maxn];
bool a[maxn][maxn], gold[maxn][maxn];

int qry() {
    vector<int> ask;
    for (int i=0;i<n;i++) for (int j=i+1;j<n;j++) if (a[i][j]) ask.push_back(id[i][j]);
    // for (int i:ask) cout << i << " "; cout << endl;
    return count_common_roads(ask);
}

int def;

int cnt(int l, int r, int pos) {
    for (int i=0;i<n;i++) for (int j=i+1;j<n;j++) a[i][j] = 0;
    for (int i=0;i<n;i++) {
        if (l <= i && i <= r) a[i][pos] = 1;
        else a[i][n-1] = 1;
    }
    int cur = qry() - def;
    for (int i=l;i<r;i++) cur += gold[i][n-1];
    return cur;
}


vector<int> find_roads(int N, vector<int> U, vector<int> V) {
    n = N;
	for (int i=0;i<n;i++) for (int j=0;j<n;j++) id[i][j] = -1, a[i][j] = 0, gold[i][j] = 0;
    for (int i=0;i<U.size();i++) {
        if (U[i] > V[i]) swap(U[i], V[i]);
        id[U[i]][V[i]] = i;
    }

    for (int i=0;i<n-2;i++) a[i][i+1] = 1;
    vector<int> vec;
    for (int i=0;i<n-1;i++) {
        a[i][n-1] = 1;
        vec.push_back(qry());
        a[i][n-1] = 0;
    }
    int mx = *max_element(vec.begin(), vec.end());
    def = 0;
    for (int i=0;i<n-1;i++) if (vec[i] == mx) {
        gold[i][n-1] = 1;
        def++;
    }
    // cout << "test\n";
    // cout << def << endl;

    for (int i=1;i<n;i++) {
        int all = cnt(0, i-1, i);
        // cout << i << " " << all << endl;
        for (int j=1;j<=all;j++) {
            int l = 0, r = i-1;
            while (l < r) {
                int mid = (l+r+1)/2;
                if (cnt(mid, i-1, i) >= j) l = mid;
                else r = mid-1;
            }
            gold[l][i] = 1;
        }
    }
    vector<int> ans;
    for (int i=0;i<n;i++) for (int j=i+1;j<n;j++) if (gold[i][j]) {
        ans.push_back(id[i][j]);
        // cout << i << " " << j << endl;
    }
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...