# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
699791 | grossly_overconfident | 버섯 세기 (IOI20_mushrooms) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "mushrooms.h"
#include <bits/stdc++.h>
using namespace std;
int count_mushrooms(int n) {
if (n < 220){
int a = 1;
for (int i = 1; i < n; ++i){
a += 1 - use_machine({ 0, i });
}
return a;
}
vector<int> a(1, 0);
vector<int> b;
if (use_machine({ 0, 1 }) == 1){
b.push_back(1);
}
else{
a.push_back(1);
}
if (use_machine({ 0, 2 }) == 1){
b.push_back(2);
}
else{
a.push_back(2);
}
for (int i = 3; i <= 150; i += 2){
if (a.size() > b.size()){
int h = use_machine({ i, a[0], i + 1, a[1]});
if (h % 2){
a.push_back(i);
}
else{
b.push_back(i);
}
if (h < 2){
a.push_back(i + 1);
}
else{
b.push_back(i + 1);
}
}
else{
int h = use_machine({ i, b[0], i + 1, b[1]});
if (h % 2){
b.push_back(i);
}
else{
a.push_back(i);
}
if (h < 2){
b.push_back(i + 1);
}
else{
a.push_back(i + 1);
}
}
}
int i = 151;
int tally = a.size();
while (true){
if (a.size() < b.size()){
vector<int> p;
int w = i;
for (int j = 0; j < b.size(); ++j){
p.push_back(i);
p.push_back(b[j]);
i++;
if (i == n){
break;
}
}
int h = use_machine(p);
if (h % 2 == 0){
b.push_back(w);
tally += h / 2;
}
else{
a.push_back(w);
tally++;
tally += (h - 1) / 2;
}
if (i == n){
return tally;
}
}
else{
vector<int> p;
int w = i;
for (int j = 0; j < a.size(); ++j){
p.push_back(i);
p.push_back(a[j]);
i++;
if (i == n){
break;
}
}
h = use_machine(p);
if (h % 2 == 0){
a.push_back(w);
tally++;
tally += (i - w) - (h / 2);
}
else{
b.push_back(w);
tally += (i - w) - ((h - 1) / 2);
}
if (i == n){
return tally;
}
}
if (i == n){
return tally;
}
}
}