BZOJ3879: SvT

题解:本想练一下虚树 可是弱鸡建不出后缀树啊 然后凉了啊  后面学了后缀树再补上后缀树的做法  现在先用单调栈+后缀数组做吧 嘤嘤嘤 具体做法同 差异那一道题

/**************************************************************
    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 MB
Submit: 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

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.

特别注意:由于另一世界线的某些参数发生了变化,对于一组询问,即使一个后缀出现了多次,也仅算一次.
posted @ 2018-08-05 17:34  wang9897  阅读(104)  评论(0编辑  收藏  举报