oj.uz

Submission #1073735

# Submission time Handle Problem Language Result Execution time Memory
1073735 2024-08-24T19:34:50 Z thatsgonzalez Counting Mushrooms (IOI20_mushrooms) C++14
0 / 100
 0 ms 344 KB 
#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;
		}
		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

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

Subtask #1
0 / 100.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Correct 0 ms 344 KB Output is correct
3 Incorrect 0 ms 344 KB Answer is not correct.
4 Halted 0 ms 0 KB -