Submission #361837

#TimeUsernameProblemLanguageResultExecution timeMemory
361837ryansee버섯 세기 (IOI20_mushrooms)C++14
0 / 100
1 ms364 KiB
#include "mushrooms.h"

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

#define FAST ios_base::sync_with_stdio(false); cin.tie(0);
#define pb push_back
#define eb emplace_back
#define ins insert
#define f first
#define s second
#define cbr cerr<<"hi\n"
#define mmst(x, v) memset((x), v, sizeof ((x)))
#define siz(x) ll(x.size())
#define all(x) (x).begin(), (x).end()
#define lbd(x,y) (lower_bound(all(x),y)-x.begin())
#define ubd(x,y) (upper_bound(all(x),y)-x.begin())
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
inline long long rand(long long x, long long y) { return rng() % (y+1-x) + x; } //inclusive
string inline to_string(char c) {string s(1,c);return s;} template<typename T> inline T gcd(T a,T b){ return a==0?llabs(b):gcd(b%a,a); }

using ll=long long; 
using ld=long double;
#define FOR(i,s,e) for(int i=s;i<=ll(e);++i)
#define DEC(i,s,e) for(ll i=s;i>=ll(e);--i)
using pi=pair<ll,ll>; using spi=pair<ll,pi>; using dpi=pair<pi,pi>; 

long long LLINF = 1e18;
int INF = 1e9+1e6;
#define MAXN (200006)

int count_mushrooms(int n) {
	int ans = 1;
	deque<int> dq, co;
	FOR(i,1,n-1) dq.eb(i);
	for(int i = 1; i < n; i += 5) {
		vector<int> tmp={0};
		FOR(j,i,min(n-1, i+4)) tmp.eb(j);
		int res = use_machine(tmp);
		if(res == 1) {
			FOR(j,1,5) if(dq.size()) dq.pop_front();
			continue;
		} else if(res == 0) {
			ans += 5;
			FOR(j,1,5) if(dq.size()) dq.pop_front();
			co.eb(i), co.eb(i+1);
			continue;
		}
		FOR(j,i,min(n-1, i+4)) if(!use_machine({0, j})) {
			co.eb(j), dq.pop_front(), ++ ans;
			if(co.size() == 2) break;
		} else dq.pop_front();
		if(co.size() == 2) break;
	}
	if(co.size() < 2) { assert(dq.empty()); return ans; }
	co.push_front(0);
	for(int i=0;i<siz(dq);i+=3) {
		vector<int> tmp;
		FOR(j,i,min((int)siz(dq)-1, i+2)) {
			tmp.eb(co[j-i]);
			tmp.eb(dq[j]);
		}
		ans += use_machine(tmp);
	}
	return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...