Submission #667151

#	Time	Username	Problem	Language	Result	Execution time	Memory
667151		Lobo	Unscrambling a Messy Bug (IOI16_messy)	C++17	100 / 100	2 ms	504 KiB

messy

#include<bits/stdc++.h>
#include "messy.h"
using namespace std;

int n;
vector<int> p;
void add(int l, int r) {
    if(l == r) {
        return;
    }
    int mid = (l+r)>>1;
    string s;
    for(int i = 0; i < n; i++) {
        if(l <= i && i <= r) s+= '0';
        else s+= '1';
    }
    for(int i = l; i <= mid; i++) {
        s[i] = '1';
        add_element(s);
        s[i] = '0';
    }
    add(l,mid);
    add(mid+1,r);
}

void check(int l, int r, set<int> st) {
    if(l == r) {
        p[*st.begin()] = l;
        return;
    }
    int mid = (l+r)>>1;

    string s;
    for(int i = 0; i < n; i++) {
        if(st.count(i)) s+= '0';
        else s+= '1';
    }
    set<int> stl,str;
    for(auto x : st) {
        s[x] = '1';
        if(check_element(s)) stl.insert(x);
        else str.insert(x);
        s[x] = '0';
    }

    check(l,mid,stl);
    check(mid+1,r,str);
}

vector<int> restore_permutation(int N, int w, int r) {
    n = N;
    add(0,n-1);
    compile_set();
    p.resize(n);
    set<int> st;
    for(int i = 0; i < n; i++) st.insert(i);
    check(0,n-1,st);
    return p;
}

• Subtask 1: 20 / 20
• Subtask 2: 18 / 18
• Subtask 3: 11 / 11
• Subtask 4: 21 / 21
• Subtask 5: 30 / 30