Submission #707008

#	Time	Username	Problem	Language	Result	Execution time	Memory
707008		NemanjaSo2005	Counting Mushrooms (IOI20_mushrooms)	C++14	56.36 / 100	10 ms	520 KiB

mushrooms

#include "mushrooms.h"
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll N,sta[20005],D=113;
bool inv;
vector<int> x,A,B,R;
stack<int> S;
int count_mushrooms(int n){
   N=n;
   if(N<=D){
      sta[0]=1;
      for(int i=0;i+1<N;i++){
         x.clear();
         x.push_back(i);
         x.push_back(i+1);
         sta[i+1]=sta[i]^use_machine(x);
      }
      ll res=0;
      for(int i=0;i<N;i++)
         res+=sta[i];
      return res;
   }
   ll res=1;
   for(int i=0;i<N;i++)
      sta[i]=2;
   sta[0]=1;
   A.push_back(0);
   for(int i=0;i+1<N;i++){
      if(A.size()>=D or B.size()>=D)
         break;
      x.clear();
      x.push_back(i);
      x.push_back(i+1);
      sta[i+1]=sta[i]^use_machine(x);
      if(sta[i+1]==1){
         A.push_back(i+1);
         res++;
      }
      else
         B.push_back(i+1);
   }
   if(A.size()>B.size()){
      R=A;
      inv=1;
   }
   else{
      R=B;
      inv=0;
   }
   for(int i=0;i<N;i++)
      if(sta[i]==2)
         S.push(i);
   int pok;
  // cout<<res<<endl;
   while(S.size()){
      x.clear();
      x.push_back(R[0]);
      pok=1;
      while(pok<R.size() and S.size()){
         x.push_back(S.top());
         S.pop();
         x.push_back(R[pok]);
         pok++;
      }
      int r=use_machine(x);
      if(inv)
         res+=((x.size()-1)-r)/2;
      else
         res+=r/2;
     // cout<<res<<endl;
   }
   return res;
}

Compilation message (stderr)

mushrooms.cpp: In function 'int count_mushrooms(int)':
mushrooms.cpp:30:18: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   30 |       if(A.size()>=D or B.size()>=D)
      |          ~~~~~~~~^~~
mushrooms.cpp:30:33: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'long long int' [-Wsign-compare]
   30 |       if(A.size()>=D or B.size()>=D)
      |                         ~~~~~~~~^~~
mushrooms.cpp:60:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   60 |       while(pok<R.size() and S.size()){
      |             ~~~^~~~~~~~~

• Subtask 1: 56.3591 / 100