Submission #897506

#	Time	Username	Problem	Language	Result	Execution time	Memory
897506		ThylOne	Rice Hub (IOI11_ricehub)	C++14	17 / 100	1061 ms	716 KiB

ricehub

#include "ricehub.h"
#include<bits/stdc++.h>

using namespace std;
vector<int> pos;
long long budget;
int limit;
bool existWithT(int T){
  int sum = 0;

  for(int i=0;i<T;i++)
    sum+=pos[i];
  
  auto computeScore = [&](int x,int l,int r){
    int re = 0;
    for(int i = l;i<=r;i++){
      re+=abs(x-pos[i]);
    }
    return re;
  };
  auto findGoodSegment = [&](int l,int r){
    int bestInf = double(sum)/double(T);
    int bestSup = ceil(double(sum)/double(T));
    
    if(computeScore(bestInf,l,r)<=budget)return true;
    if(computeScore(bestSup,l,r)<=budget)return true;
    return false;
  };
  if(findGoodSegment(0,T-1))return true;
  for(int right=T;right<pos.size();right++){
    sum-=pos[right-T];
    sum+=pos[right];
    if(findGoodSegment(right-T+1,right))return true;
  }
  return false;
}
int besthub(int R, int L, int X[], long long B)
{

  budget = B;
  pos.resize(R);
  limit = L;
  for(int i = 0;i<R;i++){
    pos[i] = X[i];
  }
  sort(pos.begin(),pos.end());
  for(int i=R;i>=1;i--){
    if(existWithT(i)){
      return i;
    }
  }
  
  
  return 1;
}

Compilation message (stderr)

ricehub.cpp: In function 'bool existWithT(int)':
ricehub.cpp:30:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |   for(int right=T;right<pos.size();right++){
      |                   ~~~~~^~~~~~~~~~~

• Subtask 1: 17 / 17
• Subtask 2: 0 / 25
• Subtask 3: 0 / 26
• Subtask 4: 0 / 32