其他分享
首页 > 其他分享> > [NOIP2020]字符串匹配 题解

[NOIP2020]字符串匹配 题解

作者:互联网

传送门QAQ

Preface

怎么题解里全是扩展 KMP 啊,好像就我不会这个东西QAQ。

只能写写大佬们都看不上的哈希+调和了>_<

Analysis

令 \(N=| S|\)。

首先发现,枚举 \(C\) 再判断前缀消耗的时间很多,这样行不通。

转向考虑枚举 \(AB\),得出所有的 \((AB)^i\),不难发现可以用哈希+调和做到 \(O(N\ln N)\)。

现在考虑 \(f(A)\le f(C)\) 的限制。

设 \(pre(i)=f(S_{1\sim i}),suf(i)=f(S_{i+1\sim n})\)。

当前枚举到 \(AB\) 的长度为 \(x\),则 \(AB\) 对答案的贡献为 \(\sum\limits_{i}\sum\limits_{j=1}^x [pre(j)\le suf(x\times i+1)]\)。

预处理出 \(pre(1\sim n),suf(1\sim n)\),用树状数组维护,时间复杂度为 \(O(TN\ln N\log N)\)。

虽然常数小,但只有 \(84\texttt{pts}\)。

仔细地思考一下,这个东西真的必须要用树状数组维护吗?

设 \(sum(i,j)= \sum\limits_{k=1}^i [pre(i)\le j]\),则 \(AB\) 的贡献变为 \(\sum\limits_{i}sum(x,x\times i+1)\)。

而 \(sum\) 数组显然可以用前缀和维护。

时间复杂度 \(O(T(N\ln N+N\times 26))\),足以通过。

Code

本地是 C++17,结果 hash 给我爆关键字了QAQ。

// Problem: #3387. 「NOIP2020」字符串匹配
// Contest: LibreOJ
// URL: https://loj.ac/p/3387
// Memory Limit: 512 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
using namespace std;
const int maxn = (1 << 20) + 5;
const int maxm = 30;
const int p = 131;
typedef long long ll;
char s[maxn];
int n,sum[maxn][maxm],Hash[maxn],pw[maxn],suf[maxn];
ll ans;
void work() {
	scanf("%s",s + 1);
	n = strlen(s + 1);
	Hash[0] = 0;
	pw[0] = 1;
	int cnt[maxm] = {0};
	int cur = 0;
	//calc pre
	for(int i = 1;i <= n;++ i) {
		for(int j = 0;j < maxm;++ j)sum[i][j] = 0;
		pw[i] = pw[i - 1] * p;
		Hash[i] = Hash[i - 1] * p + s[i];
		if(cnt[s[i] - 'a'] & 1)-- cur;
		else ++ cur;
		++ cnt[s[i] - 'a'];
		sum[i][cur] = 1;
	}
	for(int i = 1;i <= n;++ i) {
		for(int j = 1;j <= 26;++ j) {
			sum[i][j] += sum[i][j - 1];
		}
		for(int j = 0;j <= 26;++ j) {
			sum[i][j] += sum[i - 1][j];
		}
	}
	//calc suffix
	memset(cnt , 0 , sizeof(cnt));
	suf[n + 1] = 0;
	for(int i = n;i;-- i) {
		if(cnt[s[i] - 'a'] & 1)suf[i] = suf[i + 1] - 1;
		else suf[i] = suf[i + 1] + 1;
		++ cnt[s[i] - 'a'];
	}
	ans = 0;
	for(int i = 2;i < n;++ i) {
		for(int j = i;j < n;j += i) {
			if(Hash[i] != Hash[j] - Hash[j - i] * pw[i])break ;
			ans += sum[i - 1][suf[j + 1]];
		}
	}
	printf("%lld\n",ans);
	return ;
}
int main() {
	freopen("string.in","r",stdin);
	freopen("string.out","w",stdout);
	int T;
	scanf("%d",&T);
	while(T --)work();
	return 0;
}

完结撒花✿✿ヽ(°▽°)ノ✿

(回看当年赛场上瞎搞的 \(48\texttt{pts}\) 暴力感慨万千qwq)

标签:pre,AB,QAQ,limits,题解,sum,字符串,NOIP2020,sim
来源: https://www.cnblogs.com/663B/p/16548793.html