Submission #559435

#	Time	Username	Problem	Language	Result	Execution time	Memory
559435		ahmet34	Painting Squares (IOI20_squares)	C++14	0 / 100	100 ms	3056 KiB

squares

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

using ll = long long;
using pii = pair<int, int>;
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()

const int INF = 2e9, N = 1005, M = 1e9+7, LOG = 16;
const ll LINF = 1e18;

string res[N];
map<string, bool> vis;

void dfs(string node, int d) {
    for(int i = 0; i < 2; i++) {
        string to = node + char('0' + i);
        if(!vis[to]) {
            vis[to] = true;
            dfs(to.substr(1), d);
            res[d] += to.back();
        }
    }
}

map<deque<int>, int> idx[N];

void deBruijn (int d) {
	vis.clear();
	res[d].clear();
	string s = string(d-1, '0');
    dfs(s, d);
    res[d] += s;
}

int len, calculated[N];
vector<int> ans[N];

void calc(int n) {
	calculated[n] = true;
	len = min(10, n);
	deBruijn(len);

	vector<int> labels(n);
	for(int i = 0; i < n; ++i)
		labels[i] = res[len][i]-'0';

	labels.push_back(len);

	deque<int> cur (labels.begin(), labels.begin() + len);

	for(int i = 0; i < n; i++) {
		if(i+len < n) cur.push_back(labels[i+len]);
		cur.pop_front();
		idx[len][cur] = i+1;
	}
	ans[n] = labels;
}

vector<int> paint(int n) {
	len = min(10, n);
	if(!calculated[n]) calc(n);
	return ans[n];
}

int find_location(int n, vector<int> c) {
	len = min(10, n);
	if(!calculated[n]) calc(n);
	while(c.back() == -1) c.pop_back();
	return idx[min(10, n)][deque<int>(all(c))];
}

• Subtask 1: 0 / 10
• Subtask 2: 0 / 15
• Subtask 3: 0 / 20
• Subtask 4: 0 / 55