Poj3261 Milk Patterns

来源:互联网 发布:卖淘宝号会泄漏信息吗 编辑:程序博客网 时间:2024/05/09 12:44

题目传送门

题意:对一个字符串求一个最长的子串使得它至少出现k次

额,因为这个题目呢,他的字符集非常大(100W)

所以直接用SAM是不行了,我们考虑用离散化+SA,让后就可以分块rmq了

当然这样很麻烦,我们还是用SAM,但是儿子集合用map来存,这样空间就是O(n)的,时间多了一个log

#pragma GCC opitmize("O3")#pragma G++ opitmize("O3")#include<map>#include<stdio.h>#include<string.h>#include<algorithm>#define N 40005using namespace std;map<int,int> s[N];int mx[N],sz[N],f[N],ans=0;int n,m,lst=1,cnt=1,v[N],r[N];inline int extend(int c){int p=lst,np=lst=++cnt,q,nq;mx[np]=mx[p]+1; sz[np]=1;for(;p&&!s[p][c];p=f[p]) s[p][c]=np;if(!p) return f[np]=1;q=s[p][c];if(mx[q]==mx[p]+1) f[np]=q;else{nq=++cnt;mx[nq]=mx[p]+1;f[nq]=f[q]; f[q]=f[np]=nq;s[nq]=s[q];for(;p&&s[p][c]==q;p=f[p]) s[p][c]=nq;}}int main(){scanf("%d%d",&n,&m);for(int x,i=1;i<=n;++i,extend(x)) scanf("%d",&x);for(int i=1;i<=cnt;++i) ++v[mx[i]];for(int i=1;i<=n;++i) v[i]+=v[i-1];for(int i=cnt;i;--i) r[v[mx[i]]--]=i;for(int i=cnt;i;--i){sz[f[r[i]]]+=sz[r[i]];if(sz[r[i]]>=m) ans=max(ans,mx[r[i]]);}printf("%d\n",ans);}