Submission #784817

#	Time	Username	Problem	Language	Result	Execution time	Memory
784817		cadmiumsky	Ancient Books (IOI17_books)	C++17	0 / 100	8 ms	1108 KiB

books

#include <bits/stdc++.h>
#define all(x) (x).begin(),(x).end()
using namespace std;

using ll = long long;
using ld = long double;

#define int ll
#define sz(x) ((int)(x).size())

using pii = pair<int,int>;
using tii = tuple<int,int,int>;

const int nmax = 1e6 + 5, inf = 1e18 + 5;

int n;

namespace G {
  vector<vector<pii>> g;
  void init(int dim) { g.resize(dim, vector<pii>()); }
  int newnode() { g.emplace_back(); return sz(g); }
  void addedge(int x, int y, int z) {
    g[x].emplace_back(y, z);
  }
  
  namespace AINT {
    int aint[(nmax + nmax / 2) * 2];
    void init(int node = 1, int cl = 0, int cr = n - 1) {
      if(cl == cr) {
        aint[node] = cl;
        return;
      }
      int mid = cl + cr >> 1;
      init(2 * node, cl, mid);
      init(2 * node + 1, mid + 1, cr);
      aint[node] = newnode();
      addedge(aint[node], aint[2 * node], 0);
      addedge(aint[node], aint[2 * node + 1], 0);
      return;
    }
    void pushto(int val, int l, int r, int node = 1, int cl = 0, int cr = n - 1) {
      if(r < cl || cr < l) return;;
      if(l <= cl && cr <= r) {
        addedge(val, aint[node], 0);
        return;
      }
      int mid = cl + cr >> 1;
      pushto(val, l, r, 2 * node, cl, mid);
      pushto(val, l, r, 2 * node + 1, mid + 1, cr);
    }
  }
  
  priority_queue<pii, vector<pii>, greater<pii>> heap;
  vector<int> arriv, occ;
  vector<int> dijkstra(int s = -1) {
    arriv.resize(sz(g));
    occ.resize(sz(g));
    fill(all(occ), 0);
    if(s != -1) {
      fill(all(arriv), inf);
      heap.emplace(0, s);
      arriv[s] = 0;
    }
    else {
      for(int i = 0; i < sz(g); i++)
        heap.emplace(arriv[i], i);
    }
    while(!heap.empty()) {
      auto [c, node] = heap.top();
      heap.pop();
      if(occ[node]) continue;
      occ[node] = 1;
      for(auto [x, e] : g[node]) {
        if(arriv[x] > c + e) {
          arriv[x] = c + e;
          heap.emplace(arriv[x], x);
        }
      }
    }
    return arriv;
  }
}

long long minimum_walk(std::vector<signed> p, signed s) {
  n = sz(p);
  vector<int> col(n, 0);
  vector<int> rs(n + 2, 0), ls(n + 2, 0);
  int flag = 0;
  
  multiset<int> lhf;
  
  ll sum = 0;
  
  G::init(n);
  G::AINT::init();
  
  for(int i = 0; i < n; i++) {
    if(col[i] == 0) {
      flag++;
      int x = i, cnt = 0;
      set<int> vals;
      while(col[x] != flag) {
        col[x] = flag;
        sum += abs(x - p[x]);
        x = p[x];
        cnt++;
        vals.emplace(x);
      }
      int L = *vals.begin();
      int R = *vals.rbegin();
      flag++;
      x = i;
      while(col[x] != flag) {
        col[x] = flag;
        G::AINT::pushto(x, L, R);
        cerr << x << " --> " << L << ' ' << R << '\n';
        x = p[x];
      }
    }
    if(i > 0)
      G::addedge(i - 1, i, 2),
      G::addedge(i, i - 1, 2);
  }
  
  int sl = 0, sr = n - 1;
  while(p[sl] == sl && sl < s) sl++;
  while(p[sr] == sr && sr > s) sr--;
  cerr << sr << ' ' << p[sr] << '\n';
  
  auto fl = G::dijkstra(sl), fr = G::dijkstra(sr);
  for(int i = 0; i < sz(G::g); i++) {
    G::arriv[i] = fl[i] + fr[i];
  }
  auto rez = G::dijkstra();

  cerr << sum << '\n';
  
  return sum + rez[s];
}
#undef int

/**
      Anul asta se da centroid.
-- Surse oficiale
*/

Compilation message (stderr)

books.cpp: In function 'void G::AINT::init(ll, ll, ll)':
books.cpp:33:20: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   33 |       int mid = cl + cr >> 1;
      |                 ~~~^~~~
books.cpp: In function 'void G::AINT::pushto(ll, ll, ll, ll, ll, ll)':
books.cpp:47:20: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
   47 |       int mid = cl + cr >> 1;
      |                 ~~~^~~~

• Subtask 1: 0 / 12
• Subtask 2: 0 / 10
• Subtask 3: 0 / 28
• Subtask 4: 0 / 20
• Subtask 5: 0 / 30