首页 > 代码库 > BZOJ 3172 Tjoi2013 单词 后缀数组
BZOJ 3172 Tjoi2013 单词 后缀数组
题目大意:给定一个n个单词的文章,求每个单词在文章中的出现次数
文章长度<=10^6(不是单词长度<=10^6,不然读入直接超时)
首先将所有单词用空格连接成一个字符串,记录每个单词的起始位置和长度
然后求后缀数组,对于每个单词后缀数组中一定有连续一段后缀以这个单词开头,我们通过一开始记录的起始位置找到这个单词的后缀,然后左右端点二分答案,满足左右端点之间的后缀与原单词的LCP都当与等于原单词长度即可
时间复杂度O(nlogn)
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 1001001 using namespace std; int n,m; char s[M]; int st[210],len[210]; int rank[M],sa[M],height[M],X[M],Y[M]; int sum[M],cnt[M],temp[M],tot; int min_num[M][21],log2[M]; void Get_Rank() { int i; for(i=1;i<=m;i++) sum[s[i]]++; for(i=1;i<=127;i++) sum[i]+=sum[i-1]; for(i=1;i<=m;i++) sa[ sum[s[i]-1]+ ++cnt[s[i]] ]=i; for(i=1;i<=m;i++) { if(i==1||s[sa[i]]!=s[sa[i-1]]) ++tot; rank[sa[i]]=tot; } } void Radix_Sort(int key[],int order[]) { int i; for(i=0;i<=m;i++) sum[i]=cnt[i]=0; for(i=1;i<=m;i++) sum[key[i]]++; for(i=1;i<=m;i++) sum[i]+=sum[i-1]; for(i=1;i<=m;i++) temp[ sum[key[order[i]]-1]+ ++cnt[key[order[i]]] ]=order[i]; for(i=1;i<=m;i++) order[i]=temp[i]; } void Get_Height() { int i,j,k; for(i=1;i<=m;i++) { if(rank[i]==1) continue; j=max(height[rank[i-1]]-1,0);k=sa[rank[i]-1]; while(s[i+j]==s[k+j]) j++; height[rank[i]]=j; } } void Prefix_Doubling() { int i,j; Get_Rank(); for(j=1;j<=m;j<<=1) { for(i=1;i<=m;i++) { X[i]=rank[i]; Y[i]=i+j>m?0:rank[i+j]; sa[i]=i; } Radix_Sort(Y,sa); Radix_Sort(X,sa); for(i=1,tot=0;i<=m;i++) { if( i==1 || X[sa[i]]!=X[sa[i-1]] || Y[sa[i]]!=Y[sa[i-1]] ) ++tot; rank[sa[i]]=tot; } } Get_Height(); } void Input() { int i; char *p=s+1; for(i=1;i<=n;i++) { scanf("%s",p); st[i]=p-s; len[i]=strlen(p); *(p+=len[i],p++)=' '; } *(--p)=0; m=p-s-1; } inline int Min(int x,int y) { if(x>y) return 0x3f3f3f3f; return min( min_num[x][log2[y-x+1] ] , min_num[y-(1<<log2[y-x+1])+1][log2[y-x+1] ] ); } int Left_Bisection(int i) { int l=1,r=rank[st[i]]; while(l+1<r) { int mid=l+r>>1; if( Min(mid+1,rank[st[i]])>=len[i] ) r=mid; else l=mid; } if( Min(l+1,rank[st[i]])>=len[i] ) return l; return r; } int Right_Bisection(int i) { int l=rank[st[i]],r=m; while(l+1<r) { int mid=l+r>>1; if( Min(rank[st[i]]+1,mid)>=len[i] ) l=mid; else r=mid; } if( Min(rank[st[i]]+1,r)>=len[i] ) return r; return l; } int main() { int i,j; cin>>n; Input(); Prefix_Doubling(); log2[0]=-1; for(i=1;i<=m;i++) log2[i]=log2[i>>1]+1; for(i=1;i<=m;i++) min_num[i][0]=height[i]; for(j=1;j<=log2[m];j++) for(i=1;i+(1<<j)-1<=m;i++) min_num[i][j]=min( min_num[i][j-1] , min_num[i+(1<<j-1)][j-1] ); for(i=1;i<=n;i++) { int l=Left_Bisection(i); int r=Right_Bisection(i); printf("%d\n",r-l+1); } }
BZOJ 3172 Tjoi2013 单词 后缀数组
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。