题目描述:
实现 strStr() 函数。
给你两个字符串 haystack 和 needle ,请你在 haystack 字符串中找出 needle 字符串出现的第一个位置(下标从 0 开始)。如果不存在,则返回 -1 。
LeetCode原题链接
示例 1:
输入:haystack = “hello”, needle = “ll”
输出:2
示例 2:
输入:haystack = “aaaaa”, needle = “bba”
输出:-1
示例 3:
输入:haystack = “”, needle = “”
输出:0
思路:
用KMP算法,可进行子字符串的匹配,首先进行next数组的构建,为了方便,下标从1开始。
代码:
class Solution {
public:
int strStr(string s, string p) {
if(p.empty()) return 0;
int n = s.size(), m = p.size();
s = ' ' + s, p = ' ' + p; // 下标从1开始
vector<int> next(m + 1); // 关于p(needle)的next数组
// next[1]为0,因而从2开始
for(int i = 2, j = 0; i <= m; i ++ ){
while(j && p[i] != p[j + 1]) j = next[j];
if(p[i] == p[j + 1]) j ++;
next[i] = j;
}
for(int i = 1, j = 0; i <= n; i ++ ){
while(j && s[i] != p[j + 1]) j = next[j];
if(s[i] == p[j + 1]) j ++;
if(j == m) return i - m;
// i - (m - 1) - 1 = i - m,原有下标从0开始
}
return -1;
}
};
循环节大小计算
题目描述:
周期
输入格式
输入包括多组测试数据,每组测试数据包括两行。
第一行输入字符串S的长度N。
第二行输入字符串S。
输入数据以只包括一个0的行作为结尾。
输出格式
对于每组测试数据,第一行输出 “Test case #” 和测试数据的编号。
接下来的每一行,输出具有循环节的前缀的长度i和其对应K,中间用一个空格隔开。
前缀长度需要升序排列。
在每组测试数据的最后输出一个空行。
数据范围
2≤N≤1000000
输入样例:
3
aaa
4
abcd
12
aabaabaabaab
0
输出样例:
Test case #1
2 2
3 3
Test case #2
Test case #3
2 2
6 2
9 3
12 4
思路:
该题用到kmp的性质,如图所示,只考虑t整除i - ne[i]时的情况(且前后缀有交集,如图的左边两条蓝线,才能传递相等关系),由next定义可知,蓝线连接的部分相等,而又由于红线连接的部分也相等,因而可知,长度为t的部分都相等。因而,循环节的最大长度为i - ne[i].
代码:
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1000010;
int n;
int ne[N];
char str[N];
int main()
{
int T = 1;
while(scanf("%d", &n), n)
{
scanf("%s", str + 1); // 下标从1开始读
// next数组
for(int i = 2, j = 0; i <= n; i ++ )
{
while(j && str[i] != str[j + 1]) j = ne[j];
if(str[i] == str[j + 1]) j ++;
ne[i] = j;
}
printf("Test case #%d\n", T ++ );
for(int i = 1; i <= n; i ++ )
{
int t = i - ne[i]; // 长度为i时的最大循环节
// 循环节为整数,且循环节大小t不大于串长的一半
//(保证图中相等的传递,使得每个循环节都相等,即保证循环节的定义)
if(i % t == 0 && i / t >= 2)
printf("%d %d\n", i, i / t);
}
puts("");
}
return 0;
}