Submission #346266

#	Time	Username	Problem	Language	Result	Execution time	Memory
346266		casperwang	Dancing Elephants (IOI11_elephants)	C++14	97 / 100	9052 ms	7276 KiB

elephants

#include "elephants.h"
#include <bits/stdc++.h>
#define pb push_back
#define All(x) x.begin(), x.end()
#define pii pair<int,int>
#define ff first
#define ss second
using namespace std;
#define debug(args...) kout("[ " + string(#args) + " ]", args)
void kout() { cerr << endl; }
template <class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ',kout(b...); }
template <class T> void pary(T L, T R) { while (L != R) cerr << *L << " \n"[++L==R]; }

const int MAXN = 150000;
const int INF = 1e9;
const int K = 390;
int n, m, seg;
int cnt;
vector <pii> a;
vector <vector<pii>> B;
vector <pii> rng;
vector <int> pos;
vector <pii> info;

void build_block(int id, bool flag) {
  int len = B[id].size();
  if (!len) {
    rng[id] = pii(id==0 ? -INF : rng[id-1].ss, id==m-1 ? INF : rng[id+1].ff);
    return;
  }
  if (flag) sort(All(B[id]));
  int nowR = len-1;
  for (int i = len-1; i >= 0; i--) {
    pos[B[id][i].ss] = id;
    if (B[id][i].ff + seg >= B[id][len-1].ff) {
      info[B[id][i].ss].ff = 1;
      info[B[id][i].ss].ss = B[id][i].ff + seg;
    } else {
      while (nowR && B[id][i].ff + seg < B[id][nowR-1].ff) nowR--;
      info[B[id][i].ss].ff = info[B[id][nowR].ss].ff + 1;
      info[B[id][i].ss].ss = info[B[id][nowR].ss].ss;
    }
  }
  rng[id] = pii(B[id][0].ff, B[id][len-1].ff);
}

void build() {
  sort(All(a));
  for (int i = 0; i < m; i++)
    B[i].clear();
  for (int i = 0; i < n; i++)
    B[i/K].pb(a[i]);
  for (int i = 0; i < m; i++)
    build_block(i, false);
  sort(All(a), [](const pii x, const pii y) {
    return x.ss < y.ss;
  });
}

int solve() {
  int ans = 0, now = -1;
  for (int i = 0; i < m; i++) {
    auto tmp = upper_bound(All(B[i]), pii(now, n+1));
    if (tmp == B[i].end()) continue;
    ans += info[tmp->ss].ff;
    now = info[tmp->ss].ss;
  }
  return ans;
}

void init(int N, int L, int X[]) {
  cnt = 0;
  n = N, m = (n-1)/K+1, seg = L;
  a.clear(), a.resize(n);
  pos.clear(), pos.resize(n);
  info.clear(), info.resize(n);
  B.clear(), B.resize(m);
  rng.clear(), rng.resize(m);
  for (int i = 0; i < n; i++)
    a[i].ff = X[i], a[i].ss = i;
  build();
}

int update(int i, int y) {
  cnt++;
  a[i].ff = y;
  if (cnt != MAXN && cnt % (K+210) == 0) build();
  int id = pos[i];
  for (int j = 0; j < (int)B[id].size(); j++) {
    if (B[id][j].ss == i) {
      B[id].erase(B[id].begin()+j);
      break;
    }
  }
  build_block(id, false);
  int sav = -1;
  for (int j = 0; j < m; j++) {
    if ((!j || rng[j-1].ss <= y) && (j==m-1 || rng[j+1].ff >= y)) {
      if (sav == -1 || B[j].size() < B[sav].size())
        sav = j;
    }
  }
  B[sav].pb(a[i]);
  pos[i] = sav;
  build_block(sav, a[i].ff < rng[sav].ss);
  return solve();
}

• Subtask 1: 10 / 10
• Subtask 2: 16 / 16
• Subtask 3: 24 / 24
• Subtask 4: 47 / 47
• Subtask 5: 0 / 3