Problem Description
娜娜费劲九牛二虎之力终于把糖果吃完了(说好的吃不完呢?骗人,口亨~),于是,缘溪行,忘路之远近。忽逢桃花林,夹岸数百步,中无杂树,芳草鲜美,落英缤纷,娜娜甚异之。复前行,欲穷其林。林尽水源,便得一山,山有小口,仿佛若有光。便舍船,从口入。初极狭,才通人。复行数十步,豁然开朗。土地平旷,屋舍俨然,有良田美池桑竹之属。阡陌交通,鸡犬相闻。其中往来种作,男女衣着,悉如外人。黄发垂髫,并怡然自乐。(摘自《桃花源记》)
娜娜与村民交流了好久才发现这里的人们给孩子的命名方式很奇怪,首先村民们的名字都是用专门的符号来记录,正好是26个符号,于是娜娜就把它们替换成‘a’~‘z’,然后首先把爸爸的名字作为孩子的姓,妈妈的名字作为孩子的名。这时候肯定有人会问,不是独生子女怎么办?很简单~取拼接好的名字的前缀与后缀相同的部分从短到长依次作为孩子的姓名,啥,不够?那就不许你再生孩子!
不过由于桃花村民与世隔绝太久了,以致于他们总是无法正确判断一对夫妻最多能生多少个孩子,于是就把这个任务交给你了。
P.S. 若用S[1,n]表示一个长度为n的字符串,那么S[1,i](1<=i<=n)表示S的一个前缀,S[j,n](1<=j<=n)表示S的一个后缀。具体看样例解释
Input
多组数据,首先是一个正整数t(t<=20),表示数据的组数
对于每一组数据,包含两个只由'a'~'z'组成的不含空格的字符串S1,S2,分别代表丈夫的姓名以及妻子的姓名。(1<=|S1|,|S2|<=100000)
Output
Sample Input
2
ababc ababa
aaaaa aaa
Sample Output
3
8
Hint
对于样例1,把丈夫和妻子的姓名拼接在一起是ababcababa,可以作为孩子的姓名的是a、aba、ababcababa,故最多生育3个孩子
对于样例2,把丈夫和妻子的姓名拼接在一起是aaaaaaaa,可以作为孩子的姓名的是a、aa、aaa、aaaa、aaaaa、aaaaaa、aaaaaaa、aaaaaaaa,故最多生育8个孩子
题意:
给两个串a和b,合并他们成为1个串s,即s=a+b。如果串s的前缀与后缀匹配了,只要长度不同,都算可以算作一个独立的匹配,问有多少这样的匹配(注意前缀与后缀是可以重叠的)?
假设s="abcdefg",等长前缀和后缀有:len=7的abcdefg和abcdefg,len=6的abcdef和bcdefg,len=5的abcde和cdefg.....直到1个的a和g。
思路:
KMP的经典变形,这里仅有一个模式串而已,没原串。只需要对模式串求next数组就可以得到结果了,时间是O(n)。具体看下例:
假如有两串:ababc ababa
合并后变为:ababcababa
求next数组之后变为:
第1 2 3 4 5 6 7 8 9 10个
0 0 1 2 0 1 2 3 4 3
a b a b c a b a b a
找最长的合法串:第10个字符为a,而next[10]表示s[10-3+1,10]等同于s[1,3]这两个小串是相同的,也就是一个合法的名字,而且该名字是最长的(也就是长度9,因为10个的话就是自身了,毋庸置疑)。如下两个红色串:
ababcababa
接下来找次长的合法串:第next[10]个(即第3个,是a)的next应该是next[3],即1。也就是说s[1]=s[3]。而这个串在上一步才提到,是等于尾串的!看上面尾部红色的字符,aba=前部的aba,而前部aba中的后部a又等于前部的a。这说明了又是一个合法的名字。
ababcababa
接下来到next[1]=0了,也就没有再多可以匹配的了, 仅剩1个字符无法跟别人匹配。
这是模式串next数组本身的特点。细心点就可以发现。
#include <bits/stdc++.h>
#define LL long long
using namespace std;
const int N=; void get_next(string &m, vector<int> &next) //求next数组
{
next.push_back(-); //一开始是-1
int i=, j=-;
while(i<m.size())
{
if(j==- || m[i]==m[j]) //j在原串上, i在模式串上
{
next.push_back(++j);
++i;
}
else j = next[j];
}
} int cal(string &m)
{
vector<int> next;
get_next(m, next);
int len=m.size();
if(next[len]==len-) return len; //全都一样的
int i=next[len], cnt=;
while(i>) //只要next[i]>0就是一个匹配
{
cnt++;
i=next[i];//注意串s是以0开头的,而next是以1开头的。
}
return cnt+; //本身就是一个符合条件的串
} int main()
{
freopen("e://input.txt", "r", stdin);
int t;
cin>>t;
string s1,s2;
while(t--)
{
cin>>s1>>s2;
s1+=s2;
printf("%d\n",cal(s1));
}
return ;
}
AC代码