题解:本想练一下虚树 可是弱鸡建不出后缀树啊 然后凉了啊 后面学了后缀树再补上后缀树的做法 现在先用单调栈+后缀数组做吧 嘤嘤嘤 具体做法同 差异那一道题
/**************************************************************
Problem: 3879
User: c20161007
Language: C++
Result: Accepted
Time:10820 ms
Memory:70336 kb
****************************************************************/
#include <bits/stdc++.h>
const int MAXN=5e5+10;
#define ll long long
const ll mod=23333333333333333;
using namespace std;
ll read(){
ll x=0,f=1;char ch=getchar();
while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
return f*x;
}
bool cmp(int t[],int f,int w,int k){return t[f]==t[w]&&t[f+k]==t[w+k];}
int sa[MAXN],rank1[MAXN],rank2[MAXN],t1[MAXN],t2[MAXN],txt[MAXN],td[MAXN];
void Sa(char str[]){
// cout<<str<<endl;
int len=strlen(str);int m=250;
int *td=t1;int *rank1=t2;
for(int i=0;i<m;i++)txt[i]=0;
for(int i=0;i<len;i++)txt[str[i]]++,rank1[i]=str[i];
for(int i=1;i<m;i++)txt[i]+=txt[i-1];
for(int i=len-1;i>=0;i--)sa[--txt[str[i]]]=i;
for(int k=1;k<=len;k=k*2){
int p=0;
for(int i=len-k;i<len;i++)td[p++]=i;
for(int i=0;i<len;i++)if(sa[i]>=k)td[p++]=sa[i]-k;
for(int i=0;i<m;i++)txt[i]=0;
for(int i=0;i<len;i++)txt[rank1[i]]++;
for(int i=1;i<m;i++)txt[i]+=txt[i-1];
for(int i=len-1;i>=0;i--)sa[--txt[rank1[td[i]]]]=td[i];
swap(rank1,td);rank1[sa[0]]=0;
p=1;
for(int i=1;i<len;i++)rank1[sa[i]]=cmp(td,sa[i],sa[i-1],k)?p-1:p++;
if(p==len)return ;
m=p;
}
}
int H[MAXN],h[MAXN];
void hh(char str[]){
int len=strlen(str);
memset(h,0,sizeof(h));memset(H,0,sizeof(H));
for(int i=0;i<len;i++)rank2[sa[i]]=i;
for(int i=0;i<len;i++){
if(rank2[i]==0)continue;
int t=sa[rank2[i]-1];int w=i;int k;
if(i==0||H[i-1]<=1)k=0;
else k=H[i-1]-1,t+=k,w+=k;
while(t<len&&w<len){
if(str[t]==str[w])k++;
else break;
t++;w++;
}
H[i]=k;h[rank2[i]]=k;
}
}
int dp[MAXN][21];int ma[MAXN];
void St(char str[]){
ma[0]=-1;
for(int i=1;i<MAXN;i++)if((i&(i-1))==0)ma[i]=ma[i-1]+1;else ma[i]=ma[i-1];
int len=strlen(str);
for(int i=1;i<len;i++)dp[i][0]=h[i];
for(int j=1;(1<<(j-1))<=len;j++){
for(int i=1;i+(1<<j)<=len;i++){
dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
}
}
}
int rmq(int l,int r){
if(l>r)return 0;
l++;
int k=ma[r-l+1];int k1=(1<<k);
return min(dp[l][k],dp[r-k1+1][k]);
}
char s[MAXN];
bool cmp1(int aa,int bb){return rank2[aa]<rank2[bb];}
vector<int>vec;
vector<int>V;
pair<int,int> st[MAXN];int tot;
int lnum[MAXN],rnum[MAXN];
int main(){
int n,q;n=read();q=read();
scanf("%s",s);int len=strlen(s);s[len++]='$';
Sa(s);hh(s);St(s);int k,t;
for(int i=1;i<=q;i++){
scanf("%d",&k);
for(int j=1;j<=k;j++)scanf("%d",&t),t--,vec.push_back(t);
sort(vec.begin(),vec.end(),cmp1);
int sz=unique(vec.begin(),vec.end())-vec.begin();
if(sz==1){printf("0\n");continue;}
for(int j=1;j<sz;j++)V.push_back(rmq(rank2[vec[j-1]],rank2[vec[j]]));
// for(int j=0;j<V.size();j++)cout<<V[j]<<" ";
// cout<<endl;
ll ans=0;tot=0;
for(int j=0;j<V.size();j++){
while(tot>0&&V[j]<st[tot].first){tot--;}
if(!tot)lnum[j]=j+1;
else lnum[j]=j-st[tot].second;
// cout<<lnum[j]<<":::";
st[++tot]=make_pair(V[j],j);
}
//cout<<endl;
reverse(V.begin(),V.end());tot=0;
sz=V.size();
for(int j=0;j<V.size();j++){
while(tot>0&&V[j]<=st[tot].first){tot--;}
if(!tot)rnum[sz-j-1]=j+1;
else rnum[sz-j-1]=j-st[tot].second;
// cout<<rnum[j]<<":::";
st[++tot]=make_pair(V[j],j);
}
// cout<<endl;
reverse(V.begin(),V.end());
for(int j=0;j<V.size();j++)ans+=(1ll*lnum[j]*rnum[j])%mod*V[j]%mod,ans%=mod;
printf("%lld\n",ans);
for(int j=0;j<V.size();j++)lnum[j]=rnum[j]=0;
V.clear();vec.clear();
}
return 0;
}
3879: SvT
Time Limit: 30 Sec Memory Limit: 512 MBSubmit: 1275 Solved: 517
[ Submit][ Status][ Discuss]
Description
(我并不想告诉你题目名字是什么鬼)
有一个长度为n的仅包含小写字母的字符串S,下标范围为[1,n].
现在有若干组询问,对于每一个询问,我们给出若干个后缀(以其在S中出现的起始位置来表示),求这些后缀两两之间的LCP(LongestCommonPrefix)的长度之和.一对后缀之间的LCP长度仅统计一遍.
Input
第一行两个正整数n,m,分别表示S的长度以及询问的次数.
接下来一行有一个字符串S.
接下来有m组询问,对于每一组询问,均按照以下格式在一行内给出:
首先是一个整数t,表示共有多少个后缀.接下来t个整数分别表示t个后缀在字符串S中的出现位置.
Output
对于每一组询问,输出一行一个整数,表示该组询问的答案.由于答案可能很大,仅需要输出这个答案对于23333333333333333(一个巨大的质数)取模的余数.
Sample Input
7 3
popoqqq
1 4
2 3 5
4 1 2 5 6
popoqqq
1 4
2 3 5
4 1 2 5 6
Sample Output
0
0
2
Hint
样例解释:
对于询问一,只有一个后缀”oqqq”,因此答案为0.
对于询问二,有两个后缀”poqqq”以及”qqq”,两个后缀之间的LCP为0,因此答案为0.
对于询问三,有四个后缀”popoqqq”,”opoqqq”,”qqq”,”qq”,其中只有”qqq”,”qq”两个后缀之间的LCP不为0,且长度为2,因此答案为2.
对于100%的测试数据,有S<=5*10^5,且Σt<=3*10^6.
特别注意:由于另一世界线的某些参数发生了变化,对于一组询问,即使一个后缀出现了多次,也仅算一次.