Submission #898469

#	Time	Username	Problem	Language	Result	Execution time	Memory
898469		Essa2006	Counting Mushrooms (IOI20_mushrooms)	C++14	80.71 / 100	6 ms	808 KiB

mushrooms

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
#define endl '\n'
#define FF first
#define SS second
#define all(a) a.begin(), a.end()
#define mod (ll)(1000000007)
#include "mushrooms.h"
 
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
 
int count_mushrooms(int n) {
    
	vector<int> A(n);
    vector<int> Z = {0}, O;
    for (int i = 0; i < n; i++) {
        A[i] = i;
    }
    shuffle(A.begin() + 1, A.end(), rng);
    
    int ans = 1, vr = 10, last = 0;
//    for (int i = 1; i <= vr; i++){
//        last = i;
//        int cur = use_machine({Z.front(), A[i]});
//        
//        if (cur) {
//            O.push_back(A[i]);
//        }
//        else {
//            Z.push_back(A[i]);
//            ans++;
//        }
//        
//        if (n == i + 1) {
//            return ans;
//        }
//    }
    
    
 
    int i = last + 1;
    while (i < n) {
        int j = i;
        vector<int>Ask;
        if (Z.size() > O.size()) {
            while (j < n){
                if (Z.size() < j - i + 1) {
                    break;
                }
                Ask.push_back(A[j]);
                Ask.push_back(Z[j - i]);
                j++;
            }
            int diff = use_machine(Ask);
            int all = Ask.size() / 2;
            if (diff & 1) {
                int Bs = (diff + 1) / 2;
                ans += all - Bs;
                O.push_back(A[i]);
            }
            else {
                int Bs = diff / 2;
                ans += all - Bs;
                Z.push_back(A[i]);
            }
        }
        
        else {
            while (j < n){
                if (O.size() < j - i + 1) {
                    break;
                }
                Ask.push_back(A[j]);
                Ask.push_back(O[j - i]);
                j++;
            }
            int diff = use_machine(Ask);
            int all = Ask.size() / 2;
            if (diff & 1) {
                int As = (diff + 1) / 2;
                ans += As;
                Z.push_back(A[i]);
            }
            else {
                int As = diff / 2;
                ans += As;
                O.push_back(A[i]);
            }
        }
        
        i = j;
    }
    return ans;
}

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:48:30: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   48 |                 if (Z.size() < j - i + 1) {
      |                     ~~~~~~~~~^~~~~~~~~~~
mushrooms.cpp:71:30: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   71 |                 if (O.size() < j - i + 1) {
      |                     ~~~~~~~~~^~~~~~~~~~~
mushrooms.cpp:79:17: warning: unused variable 'all' [-Wunused-variable]
   79 |             int all = Ask.size() / 2;
      |                 ^~~
mushrooms.cpp:22:18: warning: unused variable 'vr' [-Wunused-variable]
   22 |     int ans = 1, vr = 10, last = 0;
      |                  ^~

• Subtask 1: 80.7143 / 100