Submission #1073738

#	Time	Username	Problem	Language	Result	Execution time	Memory
1073738		thatsgonzalez	Counting Mushrooms (IOI20_mushrooms)	C++14	80.71 / 100	7 ms	956 KiB

mushrooms

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

#define pb push_back


int count_mushrooms(int n) {
	if(n==2){
		int res = use_machine({0,1});
		if(res) return 1;
		else return 2;
	}
	int ans = 1;
	vector<int> a; a.pb(0);
	vector<int> b; 

	int res = use_machine({0,1,2});

	if(res == 0){
		ans+=2;
		a.pb(1); a.pb(2);
	}
	else if(res == 2){
		ans++;
		a.pb(2); b.pb(1);
	}
	else{
		res = use_machine({0,1});
		if(res){
			b.pb(1); b.pb(2);
		}
		else{
			ans++;
			a.pb(1); b.pb(2);
		}
	}
	int ind = 3;
	while(ind<n){
		vector <int> q; int how = 0;
		if(a.size()>=b.size()){
			int begin = ind;
			for(int i = 0; i<a.size() and ind<n; i++){
				q.pb(ind); q.pb(a[i]); ind++; how++;
			}
			res = use_machine(q);
			if(res&1){
				b.pb(begin);
			}
			else a.pb(begin);
			ans += how-(res/2+(res&1));
		}
		else{
			int begin = ind;
			for(int i = 0; i<b.size() and ind<n; i++){
				q.pb(ind); q.pb(b[i]); ind++; how++;
			}
			res = use_machine(q);
			if(res&1) a.pb(begin);
			else b.pb(begin);
			ans += res/2 + (res&1);
		}	
	}

	return ans;
}

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:43:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   43 |    for(int i = 0; i<a.size() and ind<n; i++){
      |                   ~^~~~~~~~~
mushrooms.cpp:55:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   55 |    for(int i = 0; i<b.size() and ind<n; i++){
      |                   ~^~~~~~~~~

• Subtask 1: 80.7143 / 100