Problem p02599 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <cmath>
#include <iostream>
#include <vector>

int main() {
  // 入力受け取り
  int n, q;
  std::cin >> n >> q;
  std::vector<int> c(n);
  for (int i = 0; i < n; ++i) {
    std::cin >> c[i];
    --c[i];
  }

  std::vector<int> prev(n);
  // prev[i] は， 0 <= j < i かつ c[i] == c[j] を満たす最大の j (存在しなければ
  // -1)
  {
    std::vector<int> index(n, -1);
    for (int i = 0; i < n; ++i) {
      prev[i] = index[c[i]];
      index[c[i]] = i;
    }
  }
  // デバッグ出力
  // for(int i = 0; i < n; ++i) std::cout << prev[i] << " ";
  // std::cout << std::endl;

  std::vector<int> next(n);
  // next[i] は， i < j < n かつ c[i] == c[j] を満たす最小の j (存在しなければ
  // n)
  {
    std::vector<int> index(n, n);
    for (int i = n - 1; i >= 0; --i) {
      next[i] = index[c[i]];
      index[c[i]] = i;
    }
  }
  // デバッグ出力
  // for(int i = 0; i < n; ++i) std::cout << next[i] << " ";
  // std::cout << std::endl;

  // 平方分割
  int blocksize = sqrt(n);
  int blocknum = n / blocksize + 1;
  std::vector<std::vector<int>> table(blocknum, std::vector<int>(blocknum));
  // table[i][j] (i < j) は，区間 [i * blocksize, j * blocksize) の種類数
  for (int i = 0; i < blocknum; ++i) {
    int tmp = 0;
    for (int j = i * blocksize; j <= n; ++j) {
      if (j % blocksize == 0) {
        table[i][j / blocksize] = tmp;
      }
      tmp += prev[j] < i * blocksize;
    }
  }
  // デバッグ出力
  // for(int i = 0; i < blocknum; ++i){
  //	for(int j = 0; j < blocknum; ++j){
  //		std::cout << table[i][j] << " ";
  //	}
  //	std::cout << std::endl;
  // }

  for (int i = 0; i < q; ++i) {
    int l, r;
    std::cin >> l >> r;
    --l;
    // [l, r) の種類数を答える
    int block_i = l / blocksize;
    int block_j = r / blocksize;
    int ans = table[block_i][block_j];
    for (int i = block_i * blocksize; i < l; ++i) {
      ans -= next[i] >= block_j * blocksize;
    }
    for (int i = block_j * blocksize; i < r; ++i) {
      ans += prev[i] < l;
    }
    std::cout << ans << std::endl;
  }
}


Error: TLE

Fixed submission:

#include <cmath>
#include <iostream>
#include <vector>

int main() {
  // 入力受け取り
  int n, q;
  std::cin >> n >> q;
  std::vector<int> c(n);
  for (int i = 0; i < n; ++i) {
    std::cin >> c[i];
    --c[i];
  }

  std::vector<int> next(n);
  // next[i] は， i < j < n かつ c[i] == c[j] を満たす最小の j (存在しなければ
  // n)
  {
    std::vector<int> index(n, n);
    for (int i = n - 1; i >= 0; --i) {
      next[i] = index[c[i]];
      index[c[i]] = i;
    }
  }
  // デバッグ出力
  // for(int i = 0; i < n; ++i) std::cout << next[i] << " ";
  // std::cout << std::endl;

  std::vector<int> from_zero(n + 1);
  // from_zero[i] は，区間 [0, i) の種類数
  {
    std::vector<int> count(n, 0);
    from_zero[0] = 0;
    for (int i = 0; i < n; ++i) {
      from_zero[i + 1] = from_zero[i] + !count[c[i]];
      ++count[c[i]];
    }
  }
  // デバッグ出力
  // for(int i = 0; i < n; ++i) std::cout << from_zero[i] << " ";
  // std::cout << std::endl;

  // クエリ受け取り
  for (int i = 0; i < q; ++i) {
    int l, r;
    std::cin >> l >> r;
    --l;
    // 区間[l, r) の種類数を答える
    int ans = from_zero[r];
    for (int i = 0; i < l; ++i) {
      ans -= next[i] >= r;
    }
    // なぜか答えが出る
    std::cout << ans << std::endl;
  }
}