HDU - 4552 怪盗基德的挑战书(后缀数组+RMQ/KMP+dp)

题目链接:点击查看

题目大意:给出一个字符串,统计每个前缀在字符串中出现的次数之和

题目分析:可以直接先用后缀数组跑出来height,再用RMQ跑出来任意两个后缀的height,我们可以将题意转换为求所有后缀与字符串的最长公共前缀之和,为什么可以这样转换呢,就拿样例来说,abab,其四个后缀分别为b,ab,bab,abab,首先最长的后缀也就是字符串本身,与其本身的最长公共前缀为len,就代表着前缀a,ab,aba,abab分别出现了一次,对答案的贡献为4,也就是最长公共前缀的长度,再接下来看具有贡献的后缀为ab,,代表着前缀为a和ab的字符串在后缀ab中分别出现了一次,那么对答案的贡献为2,这样问题就比较简单了,因为时间给的比较宽松,所以我就直接写了RMQ,当然也可以直接利用height的性质,因为rk[0]代表的后缀一定是原字符串,我们可以直接从rk[0]的位置分别向左向右两次维护一个最小值,每次累加这个最小值就是答案了


以上是后缀数组的做法,当然我是想练练手才用后缀数组写了一发,没想到还真能乱搞出来,更优的方法当然是KMP求出next数组然后dp做了,之前有涉及到的一个小结论就是用dp可以利用next数组线性求出每个前缀在原串中出现的次数,累加起来就是这个题目的答案了 

代码:

KMP+dp:

#include<iostream>
#include<cstdio> 
#include<string>
#include<ctime>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<cmath>
#include<sstream>
#include<unordered_map>
using namespace std;

typedef long long LL;

typedef unsigned long long ull;

const int inf=0x3f3f3f3f;

const int N=1e5+100;

char s[N];

int nx[N],dp[N],n;

void get_next()
{
	nx[0]=-1;
	int i=0,j=-1;
	while(i<n)
	{
		if(j==-1||s[i]==s[j])
			nx[++i]=++j;
		else
			j=nx[j];
	}
}

int main()
{
//	freopen("input.txt","r",stdin);
//	ios::sync_with_stdio(false);
	while(scanf("%s",s)!=EOF)
	{
		n=strlen(s);
		get_next();
		for(int i=1;i<=n;i++)
		    dp[i]=1;
		for(int i=n;i>=1;i--)
		    dp[nx[i]]+=dp[i];
		int ans=0;
		for(int i=1;i<=n;i++)
			ans+=dp[i];
		printf("%d\n",ans%256);
	}
	
	
	
	
	
	
	
	
	
	
	
	return 0;
}

后缀数组: 

#include<iostream>
#include<cstdio> 
#include<string>
#include<ctime>
#include<cstring>
#include<algorithm>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<cmath>
#include<sstream>
using namespace std;

typedef long long LL;

const int inf=0x3f3f3f3f;

const int N=1e5+100;
 
char str[N];
 
int sa[N]; //SA数组,表示将S的n个后缀从小到大排序后把排好序的
//的后缀的开头位置顺次放入SA中
int t1[N],t2[N],c[N];
 
int rk[N],height[N],len;
 
int s[N];

int st[N][20];
 
void build_sa(int s[],int n,int m)//n为添加0后的总长
{
    int i,j,p,*x=t1,*y=t2;
    for(i=0;i<m;i++) 
		c[i]=0;
    for(i=0;i<n;i++) 
		c[x[i]=s[i]]++;
    for(i=1;i<m;i++) 
		c[i]+=c[i-1];
    for(i=n-1;i>=0;i--) 
		sa[--c[x[i]]]=i;
    for(j=1;j<=n;j<<=1) 
	{
        p=0;
        for(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++) 
			c[i]=0;
        for(i=0;i<n;i++) 
			c[x[y[i]]]++;
        for(i=1;i<m;i++) 
			c[i]+=c[i-1];
        for(i=n-1;i>=0;i--) 
			sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1,x[sa[0]]=0;
        for(i=1;i<n;i++) 
            x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+j]==y[sa[i]+j]?p-1:p++;
        if(p>=n) 
			break;
        m=p;
    }
}
 
void get_height(int s[],int n)//n为添加0后的总长
{
    int i,j,k=0;
    for(i=0;i<=n;i++)
        rk[sa[i]]=i;
    for(i=0;i<n;i++) 
	{
        if(k) 
			k--;
        j=sa[rk[i]-1];
        while(s[i+k]==s[j+k]) 
			k++;
        height[rk[i]]=k;
    }
}
 
void solve(int base=128)
{
	build_sa(s,len+1,base);
	get_height(s,len);
}

void ST_build()
{
	for(int i=1;i<=len;i++)
		st[i][0]=height[i];
	for(int i=1;i<=log2(len);i++)
		for(int j=1;j+(1<<i)-1<=len;j++)
			st[j][i]=min(st[j][i-1],st[j+(1<<i-1)][i-1]);
}
 
int ST_query(int a,int b)
{
	int l=rk[a],r=rk[b];
	if(l>r)
		swap(l,r);
	l++;
	int k=log2(r-l+1);
	return min(st[l][k],st[r-(1<<k)+1][k]);
}

int main()
{
//	freopen("input.txt","r",stdin);
//	ios::sync_with_stdio(false);
	while(scanf("%s",str)!=EOF)
	{
		len=strlen(str);
		for(int i=0;i<len;i++)
			s[i]=str[i]-'a'+1;
		s[len]=0;
		solve(30);
		ST_build();
		int ans=0;
		for(int i=1;i<len;i++)
			ans+=ST_query(0,i);
		printf("%d\n",(ans+len)%256);
	}



	
	
	
	
	
	
	
	
	
	return 0;
}
发布了558 篇原创文章 · 获赞 16 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_45458915/article/details/104043130