【JZOJ6379】小w与密码(password)

description

---

analysis

• 考虑$s$的某个前缀$a$和$t$的某个前缀$b$拼成的一个字符串$a+b$

• 如果该字符串可以被另几种$a^{\prime }+b^{\prime }$拼出来，钦定$b$最短的那种只算一次

• 考虑$a_0+b_0=a_1+b_1=...=a_k+b_k$，其中$b_0$最短

• 其中满足$b_i$同时是$b_{i+1}$的前缀和后缀，可以画图理解

• 这是因为每个$b$都是$t$的前缀，且$b_i$也完全是$b_{i+1}$的后缀

• 相当于$KMP$求得的$next$的定义，$b_i$的$next$总指向$b_{i-1}$的末尾位置

• 对于$t$的前缀，只要该前缀有$next$，说明这种前缀可以被它指向的$next$的更短前缀替代

• 当然前缀还可以被它自己的$next$不断替代，直到该前缀没有$next$，即不可被替代

• 建出$t$的$next$后，拿$s$在$t$上面跑$KMP$，求出$t$每个前缀在$s$中出现的次数$num$

• 具体求就是匹配完把$num$求一次后缀和，相当于所有情况全部集中到没有$next$的前缀上，就求出了每个前缀出现次数

• 然后把答案减掉前缀出现的次数就好了

• 这题思维难但代码很短，以后要好好动脑想

---

code

#pragma GCC optimize("O3")
#pragma G++ optimize("O3")
#include
#include
#include
#define MAXN 100005
#define ll long long
#define reg register ll
#define fo(i,a,b) for (reg i=a;i<=b;++i)
#define fd(i,a,b) for (reg i=a;i>=b;--i)

using namespace std;

ll next[MAXN],num[MAXN];
char s[MAXN],t[MAXN];
ll n,m,ans;

int main()
{
	//freopen("T2.in","r",stdin);
	freopen("password.in","r",stdin);
	freopen("password.out","w",stdout);
	scanf("%s%s",s+1,t+1),n=strlen(s+1),m=strlen(t+1),ans=n*m;
	for (reg i=2,j=0;i<=m;++i)
	{
		while (j && t[j+1]!=t[i])j=next[j];
		if (t[j+1]==t[i])++j;next[i]=j;
	}
	for (reg i=2,j=0;i<=n;++i)
	{
		while (j && t[j+1]!=s[i])j=next[j];
		if (t[j+1]==s[i])
		{
			++num[++j];
			if (j==m)j=next[j];
		}
	}
	fd(i,n,1)num[next[i]]+=num[i];
	fo(i,1,m)ans-=next[i]?num[i-next[i]]:0;
	printf("%lld\n",ans);
	return 0;
}