poj 1743及spoj694

来源:互联网 发布:centos svn服务器搭建 编辑:程序博客网 时间:2024/05/22 01:45

最近在看后缀数组,刚开始看完后,自己写个后缀数组的倍增算法都累死了(ps:lz太菜了),慢慢写了几道之后好多了。

poj 1743:

题目求最长不想交子字符串,

方法是二分长度k求上限,然后根据k把height数组分组,每组字符串间的公共长度大于=k,然后对于每个组,如果sa[i]的最大值和最小值相差大于k,那么就存在两个子字符串不相交,这个k就满足题意。

#include<cstring>#include<cstdio>int n;int wa[20010],wb[20010],ws[20010],rank[20010],sa[20010],wv[20010],height[20010],r[20010];int cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}void da(int *r,int *sa,int n,int m){int *x=wa,*y=wb,*t;int  i,j,p;for(i=0;i<m;i++) ws[i]=0;for(i=0;i<n;i++)ws[x[i]=r[i]]++;for(i=1;i<m;i++)ws[i]+=ws[i-1];for(i=n-1;i>=0;i--)sa[--ws[x[i]]]=i;for(j=1,p=1;p<n;m=p,j*=2){for(p=0,i=n-j;i<n;i++)y[p++]=i;for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;for(i=0;i<m;i++)ws[i]=0;for(i=0;i<n;i++)wv[i]=x[y[i]];for(i=0;i<n;i++)ws[wv[i]]++;for(i=1;i<m;i++)ws[i]+=ws[i-1];for(i=n-1;i>=0;i--)sa[--ws[wv[i]]]=y[i];for(t=x,x=y,y=t,i=1,x[sa[0]]=0,p=1;i<n;i++)x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;}}void cal(int *r,int *sa,int n){int j;for(int i=1;i<=n;i++)rank[sa[i]]=i;int k=0;for(int i=0;i<n;height[rank[i++]]=k)for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);}int check(int x){int man=-1;int minn=30000;for(int i=0;i<n;i++){if(height[i]>=x){if(man<sa[i])man=sa[i];if(minn>sa[i])minn=sa[i];if(man-minn>=x)return 1;}else{man=minn=sa[i];}}return 0;}int main(){while(scanf("%d",&n)!=EOF){if(n==0)break;n--;int j,k;scanf("%d",&j);for(int i=0;i<n;i++){scanf("%d",&k);r[i]=k-j+100;j=k;}r[n]=0;da(r,sa,n+1,200);cal(r,sa,n);int l=1,r=n/2;while(l<r){int m=(l+r+1)>>1;if(check(m))l=m;elser=m-1;}if(l>=4)printf("%d\n",l+1);elseprintf("0\n");}}

spoj694:

题目要求所有不同的子字符串,首先还是求后缀数组,每个长度为k的后缀包含k个子字符串,然后从i=1->n计算每个后缀可以提供多少个与前面不相同的子字符串,

很明显是

n-sa[i]-height[i],最后减去的height[i]表示与rank排名前一个后缀的最长公共长度,所以又height[i]个子字符串重复了,所以,减掉,答案就是把这些值加起来即可。

#include<cstdio>#include<cstring>#define maxn 1010int T,l,wa[maxn],sa[maxn],wb[maxn],r[maxn],n,ws[maxn],height[maxn],rank[maxn];char str[1010];int cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}void da(int *r,int *sa,int n,int m){int *x=wa,*y=wb,*t;int i,j,p;for(i=0;i<m;i++)ws[i]=0;for(i=0;i<n;i++)ws[x[i]=r[i]]++;for(i=1;i<m;i++)ws[i]+=ws[i-1];for(i=n-1;i>=0;i--)sa[--ws[x[i]]]=i;for(j=1,p=1;p<n;j*=2,m=p){for(p=0,i=n-j;i<n;i++)y[p++]=i;for(i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;for(i=0;i<m;i++)ws[i]=0;for(i=0;i<n;i++)ws[x[i]]++;for(i=1;i<m;i++)ws[i]+=ws[i-1];for(i=n-1;i>=0;i--)sa[--ws[x[y[i]]]]=y[i];for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;}}void cal(int *r,int *sa,int n){int i,j;for(i=1;i<=n;i++)rank[sa[i]]=i;int k=0;for(i=0;i<n;height[rank[i++]]=k)for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);}int main(){scanf("%d",&T);while(T--){scanf("%s",str);n=strlen(str);l=0;for(int i=0;i<n;i++){r[i]=(int)str[i];}r[n]=0;da(r,sa,n+1,200);cal(r,sa,n);for(int i=1;i<=n;i++){int k=n-sa[i]-height[i];if(k>0)l+=k;}printf("%d\n",l);}return 0;}


原创粉丝点击