Submission #891368

#	Time	Username	Problem	Language	Result	Execution time	Memory
891368		stefanneagu	RMQ (NOI17_rmq)	C++17	46.10 / 100	97 ms	14148 KiB

rmq

#include <bits/stdc++.h>

using namespace std;

const int nmax = 1e5 + 1;

struct str {
  int a, b, val;
  const bool operator < (str ult) const {
    return val < ult.val;
  }
} v[nmax];

int aint[4 * nmax], lz[4 * nmax], rmq[nmax][20], used[nmax], viz[nmax];

int rmquery(int a, int b) {
  int lg = (int) log2(b - a + 1);
  return min(rmq[a][lg], rmq[b - (1 << lg) + 1][lg]);
}

void lazy(int nod, int st, int dr) {
  aint[nod] = lz[nod];
  if(st != dr) {
    lz[nod << 1] = lz[nod];
    lz[nod << 1|1] = lz[nod];
  }
  lz[nod] = 0;
}

void update(int nod, int st, int dr, int l, int r, int val) {
  if(lz[nod]) {
    lazy(nod, st, dr);
  }
  if(st > r || dr < l) {
    return;
  }
  if(l <= st && dr <= r) {
    lz[nod] = val;
    lazy(nod, st, dr);
    return;
  }
  int mid = (st + dr) / 2;
  update(nod << 1, st, mid, l, r, val);
  update(nod << 1|1, mid + 1, dr, l, r, val);
}

int query(int nod, int st, int dr, int poz) {
  if(lz[nod]) {
    lazy(nod, st, dr);
  }
  if(st == dr) {
    return aint[nod];
  }
  int mid = (st + dr) / 2;
  if(poz <= mid) {
    return query(nod << 1, st, mid, poz);
  } else {
    return query(nod << 1|1, mid + 1, dr, poz);
  }
}

int check(int n, int q, vector<int> &nou) {
  vector<vector<int>> r(n + 1, vector<int>((int) log2(n) + 4));
  for(int i = 1; i <= n; i ++) {
    r[i][0] = nou[i];
  }
  for(int j = 1; (1 << j) <= n; j ++) {
    for(int i = 1; i + (1 << j) - 1 <= n; i ++) {
      r[i][j] = min(r[i][j - 1], r[i + (1 << (j - 1))][j - 1]);
    }
  }
  for(int i = 1; i <= q; i ++) {
    int dist = (v[i].b - v[i].a + 1);
    dist = (int) log2(dist);
    int curr = min(r[v[i].a][dist], r[v[i].b - (1 << dist) + 1][dist]);
    if(curr != v[i].val) {
      return 0;
    }
  }
  return 1;
}

void bulan_pe_subtask(int n, int q, vector<int> &nou) {
  for(int i = 1; i <= n; i ++) {
    used[nou[i]] ++;
  }
  vector<int> un;
  for(int i = 0; i < n; i ++) {
    if(!used[i]) {
      un.push_back(i);
    }
  }
  for(auto nr : un) {
    bool ok = 0;
    for(int i = 1; i <= n; i ++) {
      if(!viz[i]) {
        int ult = nou[i];
        nou[i] = nr;
        if(check(n, q, nou)) {
          viz[i] = 1;
          ok = 1;
          break;
        } else {
          nou[i] = ult;
        }
      }
    }
    if(!ok) {
      for(int i = 1; i <= n; i ++) {
        cout << "-1 ";
      }
      return;
    }
  }
  for(int i = 1; i <= n; i ++) {
    cout << nou[i] << " ";
  }
}

int main() {
  int n, q;
  cin >> n >> q;
  for(int i = 1; i <= q; i ++) {
    int a, b, val;
    cin >> a >> b >> val;
    a ++;
    b ++;
    v[i].a = a, v[i].b = b, v[i].val = val;
  }
  sort(v + 1, v + q + 1);
  for(int i = 1; i <= q; i ++) {
    update(1, 1, n, v[i].a, v[i].b, v[i].val);
  }
  vector<int> nou(n + 1);
  for(int i = 1; i <= n; i ++) {
    nou[i] = query(1, 1, n, i);
    rmq[i][0] = nou[i];
  }
  for(int j = 1; (1 << j) <= n; j ++) {
    for(int i = 1; i + (1 << j) - 1 <= n; i ++) {
      rmq[i][j] = min(rmq[i][j - 1], rmq[i + (1 << (j - 1))][j - 1]);
    }
  }
  if(n <= 25) {
    bulan_pe_subtask(n, q, nou);
    return 0;
  }
  for(int i = 1; i <= q; i ++) {
    if(rmquery(v[i].a, v[i].b) != v[i].val) {
      for(int j = 1; j <= n; j ++) {
        cout << "-1 ";
      }
      return 0;
    }
  }
  for(int i = 1; i <= n; i ++) {
    cout << nou[i] << " ";
  }
  return 0;
}

• Subtask 1: 23 / 23
• Subtask 2: 13.2 / 44
• Subtask 3: 9.9 / 33