Submission #1052225

#	Time	Username	Problem	Language	Result	Execution time	Memory
1052225		TahirAliyev	Counting Mushrooms (IOI20_mushrooms)	C++17	80.43 / 100	6 ms	596 KiB

mushrooms

    #include "mushrooms.h"
    #include <bits/stdc++.h>
    using namespace std;
    #define pii pair<int, int>
    #define ll long long
    #define all(v) v.begin(), v.end()
    #define oo 1e9
    const int B = 142;
     
    int count_mushrooms(int n){
    	vector<int> v[2];
    	v[0] = {0};
    	int i = 1;
    	for(; i < min(3, n); i++){
    		v[use_machine({0, i})].push_back(i);
    	}
    	int t = (v[0].size() < v[1].size());
    	for(; i + 1 < n && (int)max(v[0].size(), v[1].size()) < B; i += 2){
    		int a = use_machine({i, v[t][0], i + 1, v[t][1]}); 
    		v[t ^ (a & 1)].push_back(i);
    		v[t ^ bool(a & 2)].push_back(i + 1);
    	}
    	if(i == n - 1){
    		v[use_machine({0, i})].push_back(i);
    		i++;
    	}
    	int ans = v[0].size();
    	if(i == n) return ans;
    	t = (v[0].size() < v[1].size());
    	while(i < n){
    		vector<int> tmp;
    		for(int j = 0; j < v[t].size() && i < n; j++, i++){
    			tmp.push_back(v[t][j]);
    			tmp.push_back(i);
    		}
    		if(t) ans += (use_machine(tmp) + 1) / 2;
    		else ans += (tmp.size() / 2) - (use_machine(tmp) + 1) / 2;
    	}
    	return ans;
    }

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:32:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |       for(int j = 0; j < v[t].size() && i < n; j++, i++){
      |                      ~~^~~~~~~~~~~~~

• Subtask 1: 80.427 / 100