[BZOJ3238][Ahoi2013]差异

Time Limit: 20 Sec
Memory Limit: 512 MB
Description

Input

一行，一个字符串S

Output

一行，一个整数，表示所求值

Sample Input

cacao

Sample Output

HINT

2<=N<=5000002<=N<=5000002<=N<=500000 , S由小写英文字母组成

题解：
学了一发后缀数组，
sa[i]=jsa[i] = jsa[i]=j表示为按照从小到大排名为i的后缀是以j（下标）开头的后缀
rk[i]=jrk[i] = jrk[i]=j表示为按照从小到大排名以i为下标开始的后缀排名为jjj
h[i]h[i]h[i]表示lcp(i,i−1)lcp(i,i-1)lcp(i,i−1)
用单调栈维护lcp(L,R)=h[i]lcp(L,R)=h[i]lcp(L,R)=h[i]重复计数就取左不取右。

#include<bits/stdc++.h>
#define LiangJiaJun main
#define ll long long
#define pa pair<int,int>
#define INF 1000000000
using namespace std;
ll ans;
int l,k,p,q,a[500004],v[500004];
int sa[2][500004],rk[2][500004],h[500004];
char str[500004];
int st[500004],L[500004],R[500004],top;
void work(int *sa,int *rk,int *SA,int *RK){for(int i=1;i<=l;i++)v[rk[sa[i]]]=i;for(int i=l;i;i--){if(sa[i]>k){SA[v[rk[sa[i]-k]]--]=sa[i]-k;}}for(int i=l-k+1;i<=l;i++)SA[v[rk[i]]--]=i;for(int i=1;i<=l;i++){RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i-1]]!=rk[SA[i]]||rk[SA[i-1]+k]!=rk[SA[i]+k]);}return ;
}
void presa(){for(int i=1;i<=l;i++)v[a[i]]++;for(int i=1;i<=30;i++)v[i]+=v[i-1];for(int i=1;i<=l;i++)sa[p][v[a[i]]--]=i;for(int i=1;i<=l;i++){rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i]]!=a[sa[p][i-1]]);}for(k=1;k<l;k<<=1,swap(p,q))work(sa[p],rk[p],sa[q],rk[q]);for(int k=0,i=1;i<=l;i++){int j=sa[p][rk[p][i]-1];while(str[j+k]==str[i+k])k++;h[rk[p][i]]=k;if(k>0)k--;}
}
void solve(){h[0]=-INF;for(int i=1;i<=l;i++)ans+=1LL*i*(l-1);top=0;st[0]=0;for(int i=1;i<=l;i++){while(h[i]<=h[st[top]])top--;if(st[top]==0)L[i]=1;else L[i]=st[top]+1;st[++top]=i;}h[l+1]=-INF;top=0;st[0]=l+1;for(int i=l;i;i--){while(h[i]<h[st[top]])top--;if(st[top]==l+1)R[i]=l;else R[i]=st[top]-1;st[++top]=i;}for(int i=1;i<=l;i++){ans-=2LL*(i-L[i]+1)*(R[i]-i+1)*h[i];}
}
int w33ha(){memset(v,0,sizeof(v));memset(h,0,sizeof(h));memset(sa,0,sizeof(sa));memset(rk,0,sizeof(rk));ans=0;p=0;q=1;l=strlen(str+1);for(int i=1;i<=l;i++)a[i]=str[i]-'a'+1;presa();solve();printf("%lld\n",ans);return 0;
}
int LiangJiaJun(){while(scanf("%s",str+1)!=EOF)w33ha();return 0;
}

[BZOJ3238][Ahoi2013]差异相关推荐

BZOJ3238[Ahoi2013]差异
BZOJ3238[Ahoi2013]差异题目描述 n<=500000n<=500000n,都是小写字母输入一行,一个字符串S 输出一行,一个整数,表示所求值 Solution 公式 ...
BZOJ3238: [Ahoi2013]差异
3238: [Ahoi2013]差异 Time Limit: 20 Sec Memory Limit: 512 MB Submit: 4840 Solved: 2298 [Submit][Stat ...
[BZOJ3238] [AHOI2013] 差异 - 后缀自动机
3238: [Ahoi2013]差异 Time Limit: 20 Sec Memory Limit: 512 MB Submit: 1968 Solved: 896 [Submit][Statu ...
bzoj3238 [Ahoi2013]差异
题目链接树状数组+单调栈计算出每个后缀的前面.后面第一个h[]比它小的(前闭后开),乘起来计算答案 1 #include<algorithm> 2 #include<iostre ...
[BZOJ3238][AHOI2013]差异 [后缀数组+单调栈]
题目地址 - GO-> 题目大意: 给定一个长度为 nn 的字符串SS,令TiTi表示它从第ii个字符开始的后缀,求以下这个式子的值: ∑1≤i<j≤nlen(Ti)+len(Tj)−2× ...
bzoj3238 [Ahoi2013]差异后缀自动机
这个题虽然看起来比较麻烦,但其实有一点搞懂了就好做了每一个后缀的前缀的lcp一定是这个串里的子串, 对于全部子串,如果它不是后缀,那一定可以找到这个串对应的后缀, 而这样的子串和后缀子串是没有区别的 ...
bzoj3238 [Ahoi2013]差异后缀自动机
题意:给出一个串,求其中任意两个字串的lcp的总和. 我们可以对于这个串建一颗后缀自动机,实际上,他的parent边树就是一颗后缀树,我们在后缀树上统计答案,设f表示right集合的大小,可以理解为后 ...
bzoj3238: [Ahoi2013]差异（后缀自动机）
传送门后缀自动机好题. 题意: 做法:samsamsam 废话考虑翻转字串,这样后缀的最长公共前缀等于前缀的最长公共后缀. 然后想到parentparentparent树上面两个串的最长公共后缀跟 ...
【bzoj3238】[Ahoi2013]差异
3238: [Ahoi2013]差异 Time Limit: 20 Sec Memory Limit: 512 MB Submit: 3425 Solved: 1559 [Submit][Stat ...

[BZOJ3238][Ahoi2013]差异

[BZOJ3238][Ahoi2013]差异相关推荐

最新文章

热门文章