【题解 && 后缀数组】 本质不同字符串

题目传送门

题目描述:

给你一个长为N的字符串,求不同的子串的个数

我们定义两个子串不同,当且仅当有这两个子串长度不一样 或者长度一样且有任意一位不一样。

子串的定义:原字符串中连续的一段字符组成的字符串


Solution

这道题是一道后缀数组的结论题。

我们尝试向 s a sa sa数组以及h数组的含义:
s a [ i ] 表 示 第 i 小 的 后 缀 串 的 左 端 点 是 什 么 sa[i]表示第i小的后缀串的左端点是什么 sa[i]i
h [ i ] b 表 示 第 i 个 串 以 及 第 i − 1 个 串 的 最 长 公 共 前 缀 h[i]b表示第i个串以及第i-1个串的最长公共前缀 h[i]bii1

枚举排序之后的字串,低于第i小的字串,它的左端点在 s a [ i ] sa[i] sa[i]这个位置上。它们一它为端点一共可以组成 n − s a [ i ] + 1 n-sa[i]+1 nsa[i]+1个串,但是如果答案一直累加上这个数的话是一定会有重复的,如何去重呢?

一个串想要与当前串重复,一定是与当前串有公共字串。
那么与 s a [ i ] sa[i] sa[i]有公共字串的串的个数很显然是 h [ i ] h[i] h[i]

答案为:
∑ i = 1 n n − s a [ i ] + 1 − h [ i ] \sum_{i=1}^n n-sa[i]+1-h[i] i=1nnsa[i]+1h[i]

为何是对的呢?
当我们枚举到第i小的串的时候,前面i-1个串是相当于已经被记录好的了,也就是说前i-1个串已经去过重,我们要思考的重复的串就是以第i个串为端点的字串的个数,所以这一波累加之后就是答案


Code

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int n;
char s[N];
int cnt[N];// ge shu
int rk[N];//pai min
int sa[N];//di i xioa de hou zhui de bian hoa
int tmp[N];//di er guan jian zi
int rkk[N];
int height[N];

signed main(){
	scanf("%d",&n);
	cin>>s+1;
	int maxx = 0;
	for (int i=1;i<=n;i++)
	  maxx = max(maxx , int(s[i]));
	for (int i=1;i<=n;i++) cnt[s[i]]++;
	for (int i=1;i<=maxx;i++) cnt[i]+=cnt[i-1];
	for (int i=n;i;i--) sa[cnt[s[i]]--] = i;
	int p;
	rk[sa[1]]=p=1;
	for (int i=2;i<=n;i++)
	  if (s[sa[i]] == s[sa[i-1]]) rk[sa[i]] = p;
	  else rk[sa[i]] = ++p;
	for (int k=1;k<=n;k<<=1){
		memset(cnt,0,sizeof cnt);
		for (int i=1;i<=n;i++) cnt[rk[i+k]] ++;//an zhao di er guan jian zi pai xu
		for (int i=1;i<=p;i++) cnt[i]+=cnt[i-1];
		for (int i=n;i;i--) tmp[cnt[rk[i+k]]--] = i;
		memset(cnt,0,sizeof cnt);
		for (int i=1;i<=n;i++) cnt[rk[i]] ++;
		for (int i=1;i<=p;i++) cnt[i]+=cnt[i-1];
		for (int i=n;i;i--) sa[cnt[rk[tmp[i]]]--] = tmp[i];//an zhao di yi guan jian zi pai xu
	    rkk[sa[1]] = p = 1;
	    for (int i=2;i<=n;i++)
	      rkk[sa[i]] = (rk[sa[i]] == rk[sa[i-1]] && rk[sa[i]+k] == rk[sa[i-1]+k] )? p:++p;//qiu chu pai xu zhi hou de pai ming
	    for (int i=1;i<=n;i++) rk[i] = rkk[i];//zhuan yi
	}
	for (int i=1,j=0;i<=n;i++){
		if (rk[i] == 1) continue;
		while (s[i+j] == s[sa[rk[i]-1]+j]) j++;
		height[rk[i]] = j;
		j = max(j-1,0);
	}
	long long ans = 0;
	for (int i=1;i<=n;i++)
	  ans+=n-sa[i]+1-height[i];
	printf("%lld",ans);
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值