Submission #481378

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

messy

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

#include "messy.h"

int n, w, r;
vector<int> res;

void add_all_elements(int l, int r){
	if(l == r) return;

	string s(n, '1');
	for(int i = l; i <= r; i++) {
		s[i] = '0';
	}
	int mid = (l + r) / 2;
	for(int i = l; i <= mid; i++){
		s[i] = '1';
		add_element(s);
		s[i] = '0';
	}
	add_all_elements(l, mid);
	add_all_elements(mid + 1, r);
}

void get_positions(int l, int r, string s){
	// cout << l << " " << r << " - " << s << endl;
	string now = s;
	if(l == r - 1){
		bool doit = false;
		for(int i = 0; i < n; i++){
			if(s[i] == '0'){
				if(doit){
					now[i] = '1';
					break;
				}
				s[i] = '1';
				if(check_element(s)){
					now[i] = '1';
					s[i] = '0';
					break;
				}
				else {
					doit = true;
				}
				s[i] = '0';
			}
		}
		for(int i = 0; i < n; i++){
			if(now[i] != s[i]){
				res[i] = l;
			}
			if(now[i] == '0'){
				res[i] = r;
			}
		}
		// cout << l << " " << r << " " << res[l] << " " << res[r] << endl;
		return;
	}
	for(int i = 0; i < n; i++){
		if(s[i] == '0'){
			s[i] = '1';
			if(check_element(s)){
				now[i] = '1';
			}
			s[i] = '0';
		}
	}
	// cout << s << " | " << now << endl;
	int mid = (l + r) / 2;
	get_positions(mid + 1, r, now);
	for(int i = 0; i < n; i++){
		if(s[i] != now[i]){
			now[i] = '0';
		}
		else {
			now[i] = '1';
		}
	}
	get_positions(l, mid, now);
}

vector<int> restore_permutation(int N, int W, int R) {
	n = N, w = W, r = R;
	res = vector<int>(n, 0);
	add_all_elements(0, n - 1);
	compile_set();
	get_positions(0, n - 1, string(n, '0'));
	return res; 
}

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