Submission #365414

#	Time	Username	Problem	Language	Result	Execution time	Memory
365414		Mlxa	Mouse (info1cup19_mouse)	C++14	48 / 100	157 ms	1652 KiB

mouse

#ifdef LC
#include "pch.h"
#else
#include <bits/stdc++.h>
#endif
using namespace std;
#define all(x) x.begin(), x.end()
#define x first
#define y second
#define mp make_pair
#define mt make_tuple

const int N = 1000;

int n;
int p[N];
int c[N][N];
bool b[N][N];
int bc[N];
int kn[N];

mt19937 rnd;

int query(vector<int>);
int qcnt = 0;

void block(int i, int j) {
	if (b[i][j]) {
		return;
	}
	b[i][j] = true;
	++bc[i];
	if (bc[i] == n - 1) {
		kn[i] = 1;
		while (b[i][kn[i]]) {
			++kn[i];
		}
		for (int ii = 0; ii < n; ++ii) {
			if (ii != i) {
				block(ii, kn[i]);
			}
		}
	}
}

bool bad(vector<int> q) {
	for (int i = 0; i < (int)q.size(); ++i) {
		if (q[i] == kn[i]) {
			return true;
		}
	}
	return false;
}

bool test(vector<int> q, int i, int j) {
	assert(query(q) == 0);
	j = (int)(find(all(q), j) - q.begin());
	swap(q[i], q[j]);
	vector<int> to;
	for (int t = 0; t < n; ++t) {
		if (t == i || t == j) {
			continue;
		}
		to.push_back(t);
	}
	shuffle(all(to), rnd);
	to.resize(2);
	bool res = false;
	for (int t : to) {
		swap(q[j], q[t]);
		if (!query(q)) {
			res = true;
		}
		swap(q[j], q[t]);
		if (res) {
			break;
		}
	}
	swap(q[i], q[j]);
	return res;
}

void solve(int _n) {
	n = _n;
	vector<int> q(n);
	iota(all(q), 1);
	for (int it = 1; it < 7 * n; ++it) {
		int cur = query(q);
		if (cur == n) {
			return;
		}
		if (cur > 1) {
			for (int i = 0; i < n; ++i) {
				c[i][q[i]] += cur;
			}
		} else if (!cur) {
			for (int i = 0; i < n; ++i) {
				block(i, q[i]);
			}
		}
		shuffle(all(q), rnd);
		while (bad(q)) {
			shuffle(all(q), rnd);
		}
		if (*min_element(bc, bc + n) == n - 1) {
			break;
		}
	}
	while (query(q)) {
		shuffle(all(q), rnd);
	}
	vector<int> ord(n);
	iota(all(ord), 0);
	for (int i : ord) {
		priority_queue<pair<int, int>> qu;
		for (int j = 1; j <= n; ++j) {
			if (!b[i][j]) {
				qu.emplace(c[i][j], j);
			}
		}
		while (qu.size() > 1) {
			int j = qu.top().y;
			if (!test(q, i, j)) {
				kn[i] = j;
				break;
			} else {
				qu.pop();
			}
		}
		if (qu.size() == 1) {
			kn[i] = qu.top().y;
		}
		for (int j = 1; j <= n; ++j) {
			if (j != kn[i]) {
				block(i, j);
			}
		}
	}
	q.assign(kn, kn + n);
	query(q);
}

#ifdef LC
int query(vector<int> q) {
	++qcnt;
	int cnt = 0;
	for (int i = 0; i < (int)q.size(); ++i) {
		cnt += p[i] == q[i];
	}
	return cnt;
}
signed main() {
	assert(freopen("input.txt", "r", stdin));
	int _n;
	cin >> _n;
	for (int i = 0; i < _n; ++i) {
		cin >> p[i];
	}
	solve(_n);
	return 0;
}
#endif

• Subtask 1: 13 / 13
• Subtask 2: 35.000014 / 38
• Subtask 3: 0 / 49