Submission #796799

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

messy

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

int n;
vector<int>ans;

void build(int l, int r)
{
    if(r-l==0) return;
    string s;
    for(int i=0; i<n; i++) s+='0';
    for(int i=l; i<=r; i++) s[i]='1';

    int mid=(l+r)>>1;
    for(int i=l; i<=mid; i++)
    {
        s[i]='0';
        add_element(s);
        // cout<<"add_element "<<s<<endl;
        s[i]='1';
    }
    build(l, mid);
    build(mid+1, r);
}

void calc(vector<int>can, int cur)
{
    if(can.size()==1)
    {
        ans[can[0]]=cur;
        return;
    }
    vector<int>v1, v2;
    string s;
    for(int i=0; i<n; i++) s+='0';
    for(int u : can) s[u]='1';
    for(int u : can)
    {
        s[u]='0';
        if(check_element(s)) v1.pb(u);
        else v2.pb(u);
        s[u]='1';
    }

    calc(v1, cur);
    calc(v2, cur+v1.size());
}

vector<int> restore_permutation(int N, int w, int r)
{
    n=N; ans.resize(n);
    build(0, n-1);
    compile_set();
    vector<int>can;
    for(int i=0; i<n; i++)
        can.pb(i);
    calc(can, 0);
    return ans;
}

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