Submission #1064931

#TimeUsernameProblemLanguageResultExecution timeMemory
1064931hyakupMeetings (IOI18_meetings)C++17
19 / 100
3449 ms7760 KiB
#include "meetings.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define bug(x) cout << #x << " " << x << endl;

const ll inf = 1e9 + 10;

const int maxn = 1e5 + 10;

vector<int> seg( 4*maxn );
void build( int pos, int ini, int fim, vector<int>& idL, vector<int>& idR ){
  if( ini == fim ){ seg[pos] = idR[ini] - idL[ini] - 1; return; }
  int l = 2*pos, r = 2*pos + 1, mid = ( ini + fim )/2;
  build( l, ini, mid, idL, idR );
  build( r, mid + 1, fim, idL, idR );
  seg[pos] = max( seg[l], seg[r] );
}

int query( int pos, int ini, int fim, int ki, int kf ){
  if( ki > fim || ini > kf ) return 0;
  if( ki <= ini && fim <= kf ) return seg[pos];
  int mid = ( ini + fim )/2;
  return max( query( 2*pos, ini, mid, ki, kf ), query( 2*pos + 1, mid + 1, fim, ki, kf ) );
}

vector<long long> minimum_costs(vector<int> H, vector<int> L, vector<int> R) {
  int n = H.size(), q = L.size();
  vector<ll> resp(q);
  if( q <= 50000 ){

    vector<ll> respL(n);
    for( int i = 0; i < q; i++ ){
      int l = L[i], r = R[i];
      resp[i] = inf*inf;

      stack<pair<int,ll>> pilha;
      pilha.push({ l - 1, inf });
      ll sum = 0;
      for( int j = l; j <= r; j++ ){
        while( pilha.top().second <= H[j] ){
          auto [id, v] = pilha.top(); pilha.pop();
          sum -= 1LL*v*abs(id - pilha.top().first);
        }
        sum += 1LL*H[j]*abs(j - pilha.top().first);
        pilha.push({ j, H[j] });
        respL[j] = sum;
      }

      while( !pilha.empty() ) pilha.pop();
      pilha.push({ r + 1, inf });
      sum = 0;
      for( int j = r; j >= l; j-- ){
        while( pilha.top().second <= H[j] ){
          auto [id, v] = pilha.top(); pilha.pop();
          sum -= 1LL*v*abs(id - pilha.top().first);
        }
        sum += 1LL*H[j]*abs(j - pilha.top().first);
        pilha.push({ j, H[j] });
        resp[i] = min( resp[i], sum + respL[j] - H[j] );
      }

    }
    return resp; 
  }
  vector<int> idL(n), idR(n);
  idL[0] = (( H[0] == 2 ) ? 0 : -1 );
  for( int i = 1; i < n; i++ ) idL[i] = (( H[i] == 2 ) ? i : idL[i - 1] );
  idR[n - 1] = (( H[n - 1] == 2 ) ? n - 1 : n );
  for( int i = n - 2; i >= 0; i-- ) idR[i] = (( H[i] == 2 ) ? i : idR[i + 1] );


  build( 1, 0, n - 1, idL, idR );

  for( int i = 0; i < q; i++ ){
    int l = L[i], r = R[i];
    resp[i] = 2*(r - l + 1) - max({ idR[l] - l, r - idL[r], query( 1, 0, n - 1, idR[l], idL[r] ) });
  }
  return resp;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...