P1117 [NOI2016]优秀的拆分 [Hash]

最新推荐文章于 2023-02-18 19:30:50 发布

FSYo

最新推荐文章于 2023-02-18 19:30:50 发布

阅读量168

点赞数

分类专栏： Hash FSY的好题汇总

本文链接：https://blog.csdn.net/sslz_fsy/article/details/95984973

版权

FSY的好题汇总同时被 2 个专栏收录

97 篇文章 0 订阅

订阅专栏

Hash

2 篇文章 0 订阅

订阅专栏

传送门

比较容易想到的是, l[i], r[i] 表示 i 的左右有多少个 AA 串

那么答案就是 $\sum l[i]*r[i+1]$ , 于是已经可以 n^2 过 95 了

我们考虑批量处理长度为 L 的AA串, 每隔 L 个位置打一个标记

然后对于两两标记之间前后求一个 LCP(Hash + 二分)

画个图, 就可以看出, 哪些区间是有贡献的, 差分一下就可以了

#include<bits/stdc++.h>
#define N 300050
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int Base = 131;
int T, n; char s[N];
ull Hash[N], Mul[N];
int l[N], r[N];
ull Get(int l, int r){
	return Hash[r] - Hash[l-1] * Mul[r - l + 1];
}
int LCS(int i, int j){
	int l = 0, r = i;
	while(l < r){
		int mid = (l+r+1) >> 1;
		if(Get(i - mid + 1, i) == Get(j - mid + 1, j)) l = mid;
		else r = mid - 1;
	} 
	return l;
}
int LCP(int i, int j){
	int l = 0, r = n - j + 1;
	while(l < r){
		int mid = (l+r+1) >> 1;
		if(Get(i, i + mid - 1) == Get(j, j + mid - 1)) l = mid;
		else r = mid - 1;
	} return l;
}
int main(){
	scanf("%d", &T);
	while(T--){
		memset(l, 0, sizeof(l));
		memset(r, 0, sizeof(r));
		scanf("%s", s+1); n = strlen(s+1);
		Mul[0] = 1;
		for(int i=1; i<=n; i++) Mul[i] = Mul[i-1] * Base;
		for(int i=1; i<=n; i++){
			Hash[i] = Hash[i-1] * Base + s[i] - 'a';
		}
		for(int L=1; L<=n*2; L++){
			for(int i=1; i<=n; i+=L){
				int j = i + L;
				int len1 = min(L, LCS(i, j)); // left
				int len2 = min(L, LCP(i, j)); // right
				if(len1 + len2 >= L){
					int res = len1 + len2 - L;
					r[i - len1 + 1]++; r[i - len1 + res + 1]--;
					l[j + len2]--; l[j + len2 - res]++;
				}
			}
		}
		for(int i=1; i<=n; i++) l[i] += l[i-1], r[i] += r[i-1];
		ll ans = 0;
		for(int i=1; i<n; i++) ans += 1ll * l[i] * r[i+1];
		printf("%lld\n", ans);
	} return 0;
}