【JZOJ6379】【NOIP2019模拟2019.10.06】小w与密码(password)

题目大意

给出两个串(S,T)，你可以通过拼接(S)的某个前缀和(T)的某个前缀得到一个新串（(S)的前缀在前），求能得到多少种不同的新串。

Solution

若一个串出现了多次，设它可以被表示成(s+t)或(s'+t')。我们令(t')比(t)更长，显然(t)是(t')的一个border。尽管(t')有多个border，但实际上只用减去最长的border带来的影响，因为更短的border已经在这个长border计算过了。考虑把(t)从(t')的后面去掉，剩下的(t')的一个前缀，若它在(S)中作为一个子串且不作为前缀出现，它就会被重复计算，减掉该前缀在(S)中出现次数即可。求前缀出现次数只需要将(T)和(S)匹配并用(next)数组做一次后缀和。

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int N = 100007;

long long ans;
int lens, lent, next[N], num[N];
char s[N], t[N];

int main() {
	freopen("password.in", "r", stdin);
	freopen("password.out", "w", stdout);
	scanf("%s%s", s + 1, t + 1);
	lens = strlen(s + 1);
	lent = strlen(t + 1);
	ans = 1ll * lens * lent;
	for (int i = 2, j = 0; i <= lent; ++i) {
		while (j && t[j + 1] != t[i]) j = next[j];
		if (t[j + 1] == t[i]) ++j;
		next[i] = j;
	}
	for (int i = 2, j = 0; i <= lens; ++i) {
		while (j && t[j + 1] != s[i]) j = next[j];
		if (t[j + 1] == s[i]) ++j;
		++num[j];
	}
	for (int i = lent; i >= 2; --i) num[next[i]] += num[i];
	for (int i = 2; i <= lent; ++i) if (next[i]) ans -= num[i - next[i]];
	printf("%lld
", ans);
	return 0;
}