結果

問題 No.2361 Many String Compare Queries
ユーザー KudeKude
提出日時 2023-06-23 23:56:15
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 83 ms / 2,500 ms
コード長 2,560 bytes
コンパイル時間 3,464 ms
コンパイル使用メモリ 250,408 KB
実行使用メモリ 16,300 KB
最終ジャッジ日時 2024-07-01 03:43:39
合計ジャッジ時間 5,107 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 79 ms
14,160 KB
testcase_09 AC 78 ms
13,748 KB
testcase_10 AC 83 ms
13,976 KB
testcase_11 AC 73 ms
13,872 KB
testcase_12 AC 82 ms
13,620 KB
testcase_13 AC 76 ms
13,748 KB
testcase_14 AC 52 ms
16,172 KB
testcase_15 AC 59 ms
16,300 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
namespace {
#pragma GCC diagnostic ignored "-Wunused-function"
#include<atcoder/all>
#pragma GCC diagnostic warning "-Wunused-function"
using namespace std;
using namespace atcoder;
#define rep(i,n) for(int i = 0; i < (int)(n); i++)
#define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--)
#define all(x) begin(x), end(x)
#define rall(x) rbegin(x), rend(x)
template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; }
template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; }
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;

int op(int x, int y) { return min(x, y); }
int e() { return 1001001001; }


template<class T, class Compare=less<T>>
pair<vector<array<int, 2>>, int> cartesian_tree(const vector<T>& a) {
  const int n = a.size();
  vector<array<int, 2>> g(n, {-1, -1});
  vector<int> st;
  constexpr Compare cmp;
  for (int i = 0; i < n; i++) {
    int last_popped = -1;
    while (!st.empty() && cmp(a[i], a[st.back()])) {
      last_popped = st.back(); st.pop_back();
    }
    g[i][0] = last_popped;
    if (!st.empty()) g[st.back()][1] = i;
    st.emplace_back(i);
  }
  return pair(move(g), st.empty() ? -1 : st[0]);
}

template<class T, class S, class Compare=less<T>>
vector<T> prefix_min_sums(const vector<S>& a) {
  int n = a.size();
  auto [to, root] = cartesian_tree<S, Compare>(a);
  vector<T> d(n + 1);
  auto dfs = [&](auto&& self, int u, int l, int r) -> void {
    if (u == -1) return;
    T v = T(a[u]) * (r - u);
    d[l] += v;
    d[u + 1] -= v;
    self(self, to[u][0], l, u);
    self(self, to[u][1], u + 1, r);
  };
  dfs(dfs, root, 0, n);
  for (int i = 0; i < n; i++) d[i + 1] += d[i];
  d.resize(n);
  return d;
}


} int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n, q;
  cin >> n >> q;
  string s;
  cin >> s;
  VI sa = suffix_array(s);
  VI lcp = lcp_array(s, sa);
  segtree<int, op, e> seg(lcp);
  VI sa_inv(n);
  rep(i, n) sa_inv[sa[i]] = i;
  VL acc(n + 1);
  rep(i, n) acc[i + 1] = acc[i] + n - sa[i];
  VL d = prefix_min_sums<ll>(lcp);
  rep(_, q) {
    int l, r;
    cin >> l >> r;
    l--;
    int len = r - l;
    int i = sa_inv[l];
    int li = seg.min_left(i, [&](int x) {
      return x >= len;
    });
    int ri = seg.max_right(i, [&](int x) {
      return x >= len;
    });
    ll ans = ll(ri - li + 1) * (len - 1);
    ans += acc[li];
    ans += d[ri];
    cout << ans << '\n';
  }
}
0