結果

問題 No.2361 Many String Compare Queries
ユーザー kotatsugame
提出日時 2023-06-23 22:28:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 440 ms / 2,500 ms
コード長 1,838 bytes
コンパイル時間 1,864 ms
コンパイル使用メモリ 105,756 KB
実行使用メモリ 43,568 KB
最終ジャッジ日時 2024-07-01 02:12:04
合計ジャッジ時間 5,531 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include<iostream>
#include<vector>
#include<algorithm>
#include<cassert>
#include<atcoder/string>
#include<atcoder/segtree>
#include<atcoder/lazysegtree>
using namespace std;
int opmin(int a,int b){return min(a,b);}
int emin(){return(int)1e9;}
using dat=pair<long long,int>;
dat opsum(dat a,dat b)
{
a.first+=b.first;
a.second+=b.second;
return a;
}
dat esum(){return make_pair(0LL,0);}
dat mp(int f,dat x)
{
x.first+=(long long)f*x.second;
return x;
}
int cmp(int f,int g){return f+g;}
int id(){return 0;}
int N,Q;
string S;
int inv[2<<17];
int L[1<<17],R[1<<17];
long long ans[1<<17];
vector<int>Qid[2<<17];
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
cin>>N>>Q>>S;
vector<int>SA=atcoder::suffix_array(S);
vector<int>LCP=atcoder::lcp_array(S,SA);
for(int i=0;i<SA.size();i++)inv[SA[i]]=i;
atcoder::segtree<int,opmin,emin>lcp_seg(LCP);
vector<dat>len_init(N);
for(int i=0;i<N;i++)len_init[i]=make_pair(N-SA[i],1);
atcoder::lazy_segtree<dat,opsum,esum,int,mp,cmp,id>len_seg(len_init);
for(int i=0;i<Q;i++)
{
cin>>L[i]>>R[i];
L[i]--;
int idx=inv[L[i]];
R[i]=R[i]-L[i];//len
int l=lcp_seg.min_left(idx,[&i](int x){return x>=R[i];});
L[i]=l;
Qid[l].push_back(i);
ans[i]=len_seg.prod(0,l).first;
}
vector<dat>init(N,make_pair(0,1));
atcoder::lazy_segtree<dat,opsum,esum,int,mp,cmp,id>seg(init);
vector<pair<int,int> >st;
for(int i=N;i--;)
{
if(i+1<N)
{
int len=LCP[i];
int cnt=0;
while(!st.empty()&&st.back().second>=len)
{
int c=st.back().first;
seg.apply(len,st.back().second,-c);
cnt+=c;
st.pop_back();
}
if(cnt>0)
{
st.push_back(make_pair(cnt,len));
}
}
seg.apply(0,N-SA[i],1);
st.push_back(make_pair(1,N-SA[i]));
for(int id:Qid[i])
{
ans[id]+=seg.prod(0,R[id]-1).first;
}
}
for(int i=0;i<Q;i++)cout<<ans[i]<<"\n";
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0