Submission #42713

#TimeUsernameProblemLanguageResultExecution timeMemory
42713funcsrRoller Coaster Railroad (IOI16_railroad)C++14
34 / 100
415 ms38580 KiB
#include "railroad.h"
#include <iostream>
#include <vector>
#include <algorithm>
#include <cassert>
using namespace std;
typedef pair<int, int> P;
#define rep(i, n) for (int i=0; i<(n); i++)
#define INF (1LL<<60)
#define INF2 1145141919
#define all(x) x.begin(), x.end()
#define uniq(x) x.erase(unique(all(x)), x.end())
#define index(x, y) (int)(lower_bound(all(x), y) - x.begin())
#define pb push_back
#define _1 first
#define _2 second
#define MAX_N (1<<19)
struct SegTree {
  P seg[MAX_N*2-1];
  SegTree() {
    rep(i, MAX_N*2-1) seg[i] = P(INF2, -1);
  }
  void update(int k, P v) {
    k += MAX_N-1;
    seg[k] = v;
    while (k > 0) {
      k = (k-1)/2;
      seg[k] = min(seg[k*2+1], seg[k*2+2]);
    }
  }
  P query(int a, int b, int k=0, int l=0, int r=MAX_N) {
    if (b <= l || r <= a) return P(INF2, -1);
    if (a <= l && r <= b) return seg[k];
    return min(query(a, b, k*2+1, l, (l+r)/2), query(a, b, k*2+2, (l+r)/2, r));
  }
} rmq;

int N;
int S[200000], T[200000];
long long dp[1<<16][16];
vector<int> G[400000];

bool f() {
  vector<int> xs;
  rep(i, N) xs.pb(S[i]), xs.pb(T[i]);
  sort(all(xs)); uniq(xs);
  rep(i, N) G[index(xs, S[i])].pb(T[i]);
  rep(i, xs.size()) sort(all(G[i]), greater<int>());
  rep(i, xs.size()) if (!G[i].empty()) rmq.update(i, P(G[i].back(), i));
  int s = 0;
  rep(_, N) {
    int p = rmq.query(s, MAX_N)._2;
    if (p == -1) return false;
    s = index(xs, G[p].back());
    G[p].pop_back();
    rmq.update(p, P(INF2, -1));
    if (!G[p].empty()) rmq.update(p, P(G[p].back(), p));
  }
  return true;
}

long long plan_roller_coaster(std::vector<int> s, std::vector<int> t) {
  N = s.size();
  rep(i, N) S[i] = s[i], T[i] = t[i];
  //if( false){
  if (N <= 16) {
    rep(i, 1<<N) rep(j, N) dp[i][j] = INF;
    rep(i, N) dp[1<<i][i] = 0;
    rep(b, 1<<N) {
      rep(last, N) {
        if (dp[b][last] == INF) continue;
        rep(t, N) if (((b>>t)&1) == 0) {
          dp[b|(1<<t)][t] = min(dp[b|(1<<t)][t], dp[b][last] + max(T[last]-S[t], 0));
        }
      }
    }
    long long m = INF;
    rep(i, N) m = min(m, dp[(1<<N)-1][i]);
    return m;
  }
  else {
    return !f();
  }
}

Compilation message (stderr)

railroad.cpp: In function 'bool f()':
railroad.cpp:8:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i, n) for (int i=0; i<(n); i++)
                                  ^
railroad.cpp:48:3: note: in expansion of macro 'rep'
   rep(i, xs.size()) sort(all(G[i]), greater<int>());
   ^
railroad.cpp:8:34: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
 #define rep(i, n) for (int i=0; i<(n); i++)
                                  ^
railroad.cpp:49:3: note: in expansion of macro 'rep'
   rep(i, xs.size()) if (!G[i].empty()) rmq.update(i, P(G[i].back(), i));
   ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...