| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 802845 | SlavicG | Rarest Insects (IOI22_insects) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "prison.h"
#include <bits/stdc++.h>
using namespace std;
#define sz(x) (int)x.size()
#define pb push_back
int n;
vector<vector<int>> st;
void rec(int idx, vector<int> elements) {
if(elements.empty()) return;
while(sz(st) <= idx) st.pb(vector<int>(n + 1));
int p = ((idx + 2) / 3) % 2;
st[idx][0] = p;
vector<int> remaining;
vector<int> A, B, C;
for(int j = 1; j <= n; ++j) {
if(j > elements.back()) st[idx][j] = (!p ? -2 : -1);
if(j < elements[0]) st[idx][j] = (!p ? -1 : -2);
}
st[idx][elements[0]] = (!p ? -1 : -2);
st[idx][elements.back()] = (!p ? -2 : -1);
elements.erase(elements.begin());
if(sz(elements)) elements.pop_back();
for(int i = 0; i < sz(elements) / 3; ++i) A.push_back(elements[i]);
for(int i = sz(elements) / 3; i < sz(elements) / 3 * 2; ++i) B.push_back(elements[i]);
for(int i = sz(elements) / 3 * 2; i < sz(elements); ++i) C.push_back(elements[i]);
if(sz(C) > sz(A)) swap(A, C);
int pp = (idx + 2) / 3 * 3;
for(auto x: A) st[idx][x] = pp + 1;
for(auto x: B) st[idx][x] = pp + 2;
for(auto x: C) st[idx][x] = pp + 3;
rec(pp + 1, A);
rec(pp + 2, B);
rec(pp + 3, C);
}
std::vector<std::vector<int>> devise_strategy(int N) {
n = N;
vector<int> elements(N);
iota(elements.begin(), elements.end(), 1);
rec(0, elements);
return st;
}