Submission #304212

#TimeUsernameProblemLanguageResultExecution timeMemory
304212juggernautCounting Mushrooms (IOI20_mushrooms)C++14
80.71 / 100
12 ms544 KiB
#include"mushrooms.h"
#include<bits/stdc++.h>
//#include"stub.cpp"
using namespace std;
int count_mushrooms(int n){
    vector<int>a,b,id;
    int i=1,res=1,j;
    for(;i<n;i++)id.push_back(i);
    shuffle(id.begin(),id.end(),default_random_engine(chrono::system_clock::now().time_since_epoch().count()));
    i=1;
    a.push_back(0);
    while(i<n){
        if(a.size()>b.size()){
            vector<int>v;
            for(j=0;j<a.size()&&i+j<n;j++){
                v.push_back(a[j]);
                v.push_back(id[i+j-1]);
            }
            i+=j;
            j=use_machine(v);
            if(j&1)b.push_back(v.back());
            else a.push_back(v.back());
            j++;
            j>>=1;
            res+=(((int(v.size()))>>1))-j;
        }else{
            vector<int>v;
            for(j=0;j<b.size()&&i+j<n;j++){
                v.push_back(b[j]);
                v.push_back(id[i+j-1]);
            }
            i+=j;
            j=use_machine(v);
            if(j&1)a.push_back(v.back());
            else b.push_back(v.back());
            j++;
            j>>=1;
            res+=j;
        }
    }
    return res;
}

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:15:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   15 |             for(j=0;j<a.size()&&i+j<n;j++){
      |                     ~^~~~~~~~~
mushrooms.cpp:28:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |             for(j=0;j<b.size()&&i+j<n;j++){
      |                     ~^~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...