結果

問題 No.2361 Many String Compare Queries
ユーザー KudeKude
提出日時 2023-06-23 23:12:30
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,528 bytes
コンパイル時間 3,088 ms
コンパイル使用メモリ 246,604 KB
実行使用メモリ 17,036 KB
最終ジャッジ日時 2023-09-13 18:20:31
合計ジャッジ時間 4,795 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 WA -
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 WA -
testcase_07 AC 2 ms
4,380 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 67 ms
14,328 KB
testcase_12 AC 77 ms
14,304 KB
testcase_13 AC 71 ms
14,292 KB
testcase_14 WA -
testcase_15 WA -
権限があれば一括ダウンロードができます

ソースコード

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]);
}


} 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(n + 1);
  {
    auto [to, root] = cartesian_tree(lcp);
    auto dfs = [&](auto&& self, int u, int l, int r) {
      if (u == -1) {
        assert(r - l == 1);
        if (l) {
          d[l] += lcp[l-1];
          d[r] -= lcp[l-1];
        }
        return;
      }
      assert(u != -1);
      ll v = ll(r - (u + 1)) * lcp[u];
      d[0] += v;
      d[u + 1] -= v;
      self(self, to[u][0], l, u + 1);
      self(self, to[u][1], u + 1, r);
    };
    dfs(dfs, root, 0, n);
    rep(i, n) d[i+1] += d[i];
  }
  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+1];
    cout << ans << '\n';
  }
}
0