首页 > 学院 > 开发设计 > 正文

bzoj 3172: [Tjoi2013]单词 后缀数组+rmq+二分

2019-11-08 01:44:19
字体:
来源:转载
供稿:网友

题意

某人读论文,一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。每个单词由小写字母组成,N<=200,文章长度不超过10^6

分析

AC自动机模板,但由于很久没打sa了想要复习一下,于是就用了sa+rmq+二分的方法来做。

代码

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>#include<cmath>#define N 1010005using namespace std;int sa[N],rank[N],height[N],b[N],c[N],d[N],m,pos[205],rmq[N][30],s[N],len[205];char ch[N];void getsa(int n,int m){ for (int i=0;i<=m;i++) b[i]=0; for (int i=1;i<=n;i++) b[s[i]]++; for (int i=1;i<=m;i++) b[i]+=b[i-1]; for (int i=n;i>=1;i--) c[b[s[i]]--]=i; int t=0; for (int i=1;i<=n;i++) { if (s[c[i]]!=s[c[i-1]]) t++; rank[c[i]]=t; } int j=1; while (j<=n) { for (int i=0;i<=n;i++) b[i]=0; for (int i=1;i<=n;i++) b[rank[i+j]]++; for (int i=1;i<=n;i++) b[i]+=b[i-1]; for (int i=n;i>=1;i--) c[b[rank[i+j]]--]=i; for (int i=0;i<=n;i++) b[i]=0; for (int i=1;i<=n;i++) b[rank[i]]++; for (int i=1;i<=n;i++) b[i]+=b[i-1]; for (int i=n;i>=1;i--) d[b[rank[c[i]]]--]=c[i]; int t=0; for (int i=1;i<=n;i++) { if (rank[d[i]]!=rank[d[i-1]]||rank[d[i]]==rank[d[i-1]]&&rank[d[i]+j]!=rank[d[i-1]+j]) t++; c[d[i]]=t; } for (int i=1;i<=n;i++) rank[i]=c[i]; if (t==n) break; j*=2; } for (int i=1;i<=n;i++) sa[rank[i]]=i;}void getheight(int n){ int k=0; for (int i=1;i<=n;i++) { if (k) k--; int j=sa[rank[i]-1]; while (i+k<=n&&j+k<=n&&s[i+k]==s[j+k]) k++; height[rank[i]]=k; }}void get_rmq(int n){ int lg=log(n)/log(2); for (int i=1;i<=n;i++) rmq[i][0]=height[i]; for (int j=1;j<=lg;j++) for (int i=1;i<=n-(1<<j)+1;i++) rmq[i][j]=min(rmq[i][j-1],rmq[i+(1<<(j-1))][j-1]);}int get_min(int l,int r){ int lg=log(r-l+1)/log(2); return min(rmq[l][lg],rmq[r-(1<<lg)+1][lg]);}int efl(int pos,int len,int n){ int l=1,r=pos-1; while (l<=r) { int mid=(l+r)/2; if (get_min(mid+1,pos)>=len) r=mid-1; else l=mid+1; } return r+1;}int efr(int pos,int len,int n){ int l=pos+1,r=n; while (l<=r) { int mid=(l+r)/2; if (get_min(pos+1,mid)>=len) l=mid+1; else r=mid-1; } return l-1;}int main(){ scanf("%d",&m); int n=0,tot=123; for (int i=1;i<=m;i++) { scanf("%s",ch); len[i]=strlen(ch);pos[i]=n+1; for (int j=0;j<len[i];j++) s[++n]=ch[j]; s[++n]=tot++; } getsa(n,tot); getheight(n); get_rmq(n); for (int i=1;i<=m;i++) { int l=efl(rank[pos[i]],len[i],n),r=efr(rank[pos[i]],len[i],n);; PRintf("%d/n",r-l+1); } return 0;}
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表