洛谷 P3375 【模板】KMP字符串匹配

题目描述

给出两个字符串 s_1s1​ 和 s_2s2​，若 s_1s1​ 的区间 [l, r][l,r] 子串与 s_2s2​ 完全相同，则称 s_2s2​ 在 s_1s1​ 中出现了，其出现位置为 ll。
现在请你求出 s_2s2​ 在 s_1s1​ 中所有出现的位置。

定义一个字符串 ss 的 border 为 ss 的一个非 ss 本身的子串 tt，满足 tt 既是 ss 的前缀，又是 ss 的后缀。
对于 s_2s2​，你还需要求出对于其每个前缀 s's′ 的最长 border t't′ 的长度。

输入格式

第一行为一个字符串，即为 s_1s1​。
第二行为一个字符串，即为 s_2s2​。

输出格式

首先输出若干行，每行一个整数，按从小到大的顺序输出 s_2s2​ 在 s_1s1​ 中出现的位置。
最后一行输出 |s_2|∣s2​∣ 个整数，第 ii 个整数表示 s_2s2​ 的长度为 ii 的前缀的最长 border 长度。

输入输出样例

输入 #1复制

ABABABC
ABA

输出 #1复制

1
3
0 0 1 

说明/提示

样例 1 解释

。

对于 s_2s2​ 长度为 33 的前缀 ABA，字符串 A 既是其后缀也是其前缀，且是最长的，因此最长 border 长度为 11。

数据规模与约定

本题采用多测试点捆绑测试，共有 3 个子任务。

• Subtask 1（30 points）：|s_1| \leq 15∣s1​∣≤15，|s_2| \leq 5∣s2​∣≤5。
• Subtask 2（40 points）：|s_1| \leq 10^4∣s1​∣≤104，|s_2| \leq 10^2∣s2​∣≤102。
• Subtask 3（30 points）：无特殊约定。

对于全部的测试点，保证 1 \leq |s_1|,|s_2| \leq 10^61≤∣s1​∣,∣s2​∣≤106，s_1, s_2s1​,s2​ 中均只含大写英文字母。

一开始是去B站看的正月点灯笼的视频，之前并查集也是在那看的，算法思路和代码实现过程都很清楚，相对于别的博客和视频算是比较容易懂的，视频传送门

提一下 ：视频里是用int *prefix=malloc(sizeof(int * n)，但是编译器会报错，C/C++前面必须要加类型int *prefix=（int*)malloc(sizeof(int * n),虽然但是我最后过的代码里没有这行

 昨天错的真的是多姿多彩 

 

 把数组开大后移到前面作全局变量还是没过，还是上面一样的报错

错误代码：

#include<stdio.h>
#include<stdlib.h>
#include<string.h>

	char s1[1000020];
	char s2[1000010];
	int next[1000010];

void prefix_table(char s2[],int next[],int n)//求出前缀表
{
	next[0]=0;
	int len=0;
	int i=1;
	while(i<n)
	{
		if(s2[i]==s2[len])
		{
			len++;
			next[i]=len;
			i++;
		}
		else
		{
			if(len>0)
			{
				len=next[len-1];
			}
			else
			{	
				next[i]=len;
				i++;
			}
		}
	}
}

void kmp_search(char s1[],char s2[])
{
	int m=strlen(s1);
	int n=strlen(s2);
	prefix_table(s2,next,n);
	printf("n=%d\n",n);
	
	int i=0,j=0;
	while(i<m)
	{
		if(j==n-1&&s1[i]==s2[j])//n-1因为数组从0开始，s2最后一个字符都对上了，搜索到字符串出现的位置 
		{
			//printf("i=%d,j=%d ",i,j); 
			printf("%d\n",i-j+1);//printf("位置：%d\n",i-j+1);
			//s2最后一个字符位置j与对应s1字符位置i，i-j+1得出s2首字符位置 
			i=i+1-next[j];
			j=0;
		}
		
		if(s1[i]==s2[j])//字符依次对应 
		{
			//printf("move i=%d,j=%d\n",i,j);
			i++,j++;
		}
		else//对不上的时候根据前缀表移动 
		{
			j=next[j];//将s2字符串，按前缀表下标，将s2字符数组对应序号，移到当前位置 
			//printf("前缀move i=%d,j=%d\n",i,j);
			if(j==0)
			{
				i++,j++;
			}
		}
	}
}

int main()
{
	scanf("%s%s",s1,s2);
	
	int n=strlen(s2);
	prefix_table(s2,next,n);//求前缀表 
	kmp_search(s1,s2);//搜索位置 
	
	for(int i=0;i<n;i++)
	{
		printf("%d ",next[i]);//打印前缀表，即border长度 
	}
	
	return 0;
}

今天早上起来再改，第四遍终于过了

AC代码：

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>

char s1[1000020];
char s2[1000010];
int next[1000010];
void prefix_table(char s2[],int next[],int n)//求出前缀表
{
	next[0]=0;
	int len=0;
	int i=1;
	while(i<n)
	{
		if(s2[i]==s2[len])
		{
			len++;
			next[i]=len;
			i++;
		}
		else
		{
			if(len>0)
			{
				len=next[len-1];
			}
			else
			{	
				next[i]=len;
				i++;
			}
		}
	}
}

void kmp(char s1[],char s2[],int n,int m)
{
    int j=0;
    for(int i=0;i<m;i++)
    {
        while(j>0&&s2[j]!=s1[i])
			j=next[j-1]; //回跳
        if (s2[j]==s1[i])
			j++;
        if (j==n)
        {
            printf("%d\n",i+1-n+1);
            j=next[j-1];//继续匹配
        }
    }
}

int main()
{
    scanf("%s%s",&s1,&s2);
    int n=strlen(s2);
    int m=strlen(s1);
    prefix_table(s2,next,n);
    kmp(s1,s2,n,m);
    for(int i=0;i<n;i++)
    {
        printf("%d ",next[i]);
    }
    return 0;
}