JZOJ.5315【NOIP2017模拟8.19】小串串

Description

 

Input

Output

 

Sample Input

4
aa
abcd
ccc
abcc

Sample Output

5
10
14
12

 

Data Constraint

字符串操作我们可以采用任何后缀的东西，包括后缀数组，后缀自动机，后缀树等等...

鉴于某蒟蒻只会后缀数组下面我们采用后缀数组的方法

本题要求出某字符子串在原串出现的次数，我们对原串求出height数组后，将height[i]值作为纵坐标，横坐标为i我们可以发现

矩形宽度即为某子串出现的次数，高度为该串的长度，每个单位高度的矩形就对答案贡献宽度的平方值，于是我们用单调栈来统计这矩形对答案的贡献，然后剩下的子串对答案都贡献1即可。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#define N 100005
using namespace std;
long long t,n,sa[N],rank[N],tp[N],tax[N],h[N],len,a[N],top,id[N],sum,m,ans,oo[N];
char qwq[N];
void rsort(){
    for (int i=1;i<=m;i++) tax[i]=0;
    for (int i=1;i<=len;i++) tax[rank[tp[i]]]++;
    for (int i=1;i<=m;i++) tax[i]+=tax[i-1];
    for (int i=len;i>=1;i--) sa[tax[rank[tp[i]]]--]=tp[i];
}
bool comp(long long *f,int x,int y,int w){
    return ((f[x]==f[y])&&(f[x+w]==f[y+w]));
}
void SS(){
    for (int i=1;i<=len;i++) rank[i]=a[i],tp[i]=i;
    m=128;rsort();
    for (int i,p=1,w=1;p<len;w+=w,m=p){
        for (p=0,i=len-w+1;i<=len;i++) tp[++p]=i;
        for (i=1;i<=len;i++) if (sa[i]>w) tp[++p]=sa[i]-w;
        rsort();swap(tp,rank);rank[sa[1]]=1;p=1;
    for (i=2;i<=len;i++) rank[sa[i]]=comp(tp,sa[i],sa[i-1],w)?p:++p;
    }
    int j=0,k=0;
    for (int i=1;i<=len;oo[rank[i++]]=k)
     for (k=k?k-1:k,j=sa[rank[i]-1];a[i+k]==a[j+k];k++);
}
void work(){
    long long tmp=0;
    for (long long i=1;i<=len;i++){
        while ((h[id[top]]>=h[i])&&(top>0)){
            long long x=i-id[top-1];
            long long y=h[id[top]]-max(h[id[top-1]],h[i]);
            ans+=x*x*y;
            sum+=x*y;
            top--;
        }
        id[++top]=i;
    }
    ans+=(n-sum);
    printf("%lld\n",ans);
}
void clear(){
    memset(sa,0,sizeof(sa));
    memset(rank,0,sizeof(rank));
    memset(h,0,sizeof(h));
    memset(id,0,sizeof(id));
    memset(a,0,sizeof(a));
    top=0;
    ans=0;
    sum=0;
    len=0;
    n=0;
}
int main(){
    for (scanf("%lld",&t);t;t--){
        scanf("%s",qwq);
        n=strlen(qwq);
        for (int i=0;i<n;i++)
            a[i+1]=(long long)qwq[i]-(long long)'a'+1;
        len=n;
        n=n*(n+1)/2;
        SS();
        for (int i=1;i<=len;i++)
         h[i]=oo[i+1];
        work();
        clear();
    }
    return 0;
}

第一次打单调栈打到崩溃QAQ