Submission #1211969

#	Time	Username	Problem	Language	Result	Execution time	Memory
1211969		mychecksedad	Roller Coaster Railroad (IOI16_railroad)	C++20	30 / 100	174 ms	25480 KiB

railroad

#include "railroad.h"
#include <bits/stdc++.h>
using namespace std;
#define ll long long int
#define MOD (1000000000+7)
#define MOD1 (998244353)
#define pb push_back
#define all(x) x.begin(), x.end()
#define en cout << '\n'
#define ff first
#define ss second
#define pii pair<int,int>
#define vi vector<int>
const int N = 1e6+100, M = 1e5+10, K = 52, MX = 30;

long long plan_roller_coaster(std::vector<int> s, std::vector<int> t) {
  int n = s.size();
  vector<array<ll, 3>> A, B;
  multiset<pair<ll, int>> T;
  for(int i = 1; i <= n; ++i){
    A.pb({s[i-1], t[i-1], i-1});
    B.pb({t[i-1], s[i-1], i-1});
    T.insert({s[i-1], i-1});
  }
  T.insert({MOD, n+1});
  sort(all(A));
  sort(all(B));

  ll ans = 0;
  for(int i = n-1; i >= 0; --i){
    int pos = B[i][2];
    auto it = T.lower_bound(pair<ll,int>{B[i][0], -1});
    if(it == T.end()){
      --it;
      if((*it).ss == pos){
        --it;
      }
      ans += B[i][0] - (*it).ff;
      // cerr << (*it).ff << ' ';
      T.erase(it);
    }else{
      if((*it).ss == pos){
        if(next(it) == T.end()){
          --it;
        }else{
          ++it;
        }
        // cerr << (*it).ff << ' ';
        ans += max(0ll, B[i][0] - (*it).ff);
        T.erase(it);
      }else{
        ans += max(0ll, B[i][0] - (*it).ff);
        // cerr << (*it).ff << ' ';
        T.erase(it);
      }
    }
  }

  return ans;
}

Compilation message (stderr)

railroad.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
railroad_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

• Subtask 1: 0 / 11
• Subtask 2: 0 / 23
• Subtask 3: 30 / 30
• Subtask 4: 0 / 36