結果

問題 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,347 ms
コンパイル使用メモリ 249,108 KB
実行使用メモリ 16,940 KB
最終ジャッジ日時 2024-07-01 02:58:43
合計ジャッジ時間 4,834 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 WA -
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 WA -
testcase_07 AC 2 ms
6,940 KB
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 AC 73 ms
14,380 KB
testcase_12 AC 79 ms
14,512 KB
testcase_13 AC 75 ms
14,508 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';
}
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0