Z函数学习笔记(扩展KMP)

原创已于 2022-03-13 07:48:40 修改 · 1.9k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#学习 #算法 #哈希算法

于 2021-07-26 09:49:50 首次发布

字符串专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了Z函数的概念及其在字符串匹配中的应用，包括Z函数的基本定义、ZBox的概念、ex-KMP算法实现，以及如何通过Z函数求解字符串的最长公共前缀等问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ex-KMP(Z函数)

~~下标全部从1开始~~

Z函数可以解决的问题：

$线性求解字符串 s t r 以第 i 位开始的后缀与 s t r 的最长公共前缀 (l c p)$
$Z [i] 表示第 i 位开始的后缀与 s t r 的最长公共前缀$

Z Box引入

$\ Box是字符串str中的一个区间[l,r]满足str[l,r]是str的前缀(不一定要求最长)，随i移动而变化$
$在位置 i 时， [l, r] 必须包含 i, r 要尽可能的大$

在这里插入图片描述

ex-KMP写法

如何计算 $Z$ 数组(真的和KMP非常像)，因为有 $Z [1] = n$ ,利用递推的思想，所以递推求解，假设 $已知i-1位置的Z\ Box 的情况下我们来计算Z[i],和i位置的Z\ Box$ .
分为三种情况

时间复杂度

求Z[i]，需要从1~n遍历,O(n)
Z Box右端点最多右移n次,O(n)
总体复杂度O(n)

参考b站Z函数的代码
void Z(int n,char *str){
   int l=0,r=0;//定义Z box
   Z[1]=n;
   for(int i=2;i<=n;i++){
       if(i>r){//超过了Z box范围
           while(str[i+Z[i]]==str[1+Z[i]])
               	Z[i]++;
          	l=i,r=i+Z[i]-1;
       }
       else if(Z[i-l+1]<r-i+1) //O(1)
           Z[i]=Z[i-l+1];
       else{
           Z[i]=r-i;
           while(str[i+Z[i]]==str[1+Z[i]])
               Z[i]++;
           l=i,r=i+Z[i]-1;
       }
   }
}

Z函数扩展有扩展KMP

给出两个字符串str1,str2,求str1的每一个后缀与str2的最长公共前缀
$O (n)$
$设 e x t [i] 表示 s t r 1 第 i 位开始的后缀和 s t r 2 的最长公共前缀 (l c p)$
$\ box开一个ext\ box，$

参考b站ext-kmp函数的代码
void get_ext(int n,char *str1,int m,char *str2){
	int l=0,r=0;//定义Z box

    for(int i=1;i<=n;i++){
        if(i>r){//超过了Z box范围
            while(1+ext[i]<=m&&i+ext[i]<=n&&str1[i+ext[i]]==str2[1+ext[i]])
                	ext[i]++;
           	l=i,r=i+ext[i]-1;
        }
        else if(Z[i-l+1]<r-i+1) //O(1)
            ext[i]=Z[i-l+1];
        else{
            ext[i]=r-i;
            while(1+ext[i]<=m&&i+ext[i]<=n&&str1[i+ext[i]]==str2[1+ext[i]])
                ext[i]++;
            l=i,r=i+ext[i]-1;
        }
    }
}

Problem:P5410 【模板】扩展 KMP（Z 函数）

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <queue>
#include <set>
#include <map>
#include <vector>
#include <sstream>
#define pb push_back 
#define in insert
#define mem(f, x) memset(f,x,sizeof(f)) 
#define fo(i,a,n) for(int i=(a);i<=(n);++i)
#define fo_(i,a,n) for(int i=(a);i<(n);++i)
#define debug(x) cout<<#x<<":"<<x<<endl;
#define endl '\n'
using namespace std;

template<typename T>
ostream& operator<<(ostream& os,const vector<T>&v){for(int i=0,j=0;i<v.size();i++,j++)if(j>=5){j=0;puts("");}else os<<v[i]<<" ";return os;}
template<typename T>
ostream& operator<<(ostream& os,const set<T>&v){for(auto c:v)os<<c<<" ";return os;}

typedef pair<int,int>PII;
typedef pair<long,long>PLL;

typedef long long ll;
typedef unsigned long long ull; 
const int N=2e7+10;
ll n,m,_;

int Z[N],ext[N];
char a[N],b[N];

void get_Z(int n,char *str){
	int l=0,r=0;//定义Z box
    Z[1]=n;
    for(int i=2;i<=n;i++){
        if(i>r){//超过了Z box范围
            while(str[i+Z[i]]==str[1+Z[i]])
                Z[i]++;
           	l=i,r=i+Z[i]-1;
        }
        else if(Z[i-l+1]<r-i+1) //O(1)
            Z[i]=Z[i-l+1];
        else{
            Z[i]=r-i;
            while(str[i+Z[i]]==str[1+Z[i]])
                Z[i]++;
            l=i,r=i+Z[i]-1;
        }
    }
}

void get_ext(int n,char *str1,int m,char *str2){

	int l=0,r=0;//定义Z box

    for(int i=1;i<=n;i++){
        if(i>r){//超过了Z box范围
            while(1+ext[i]<=m&&i+ext[i]<=n&&str1[i+ext[i]]==str2[1+ext[i]])
                	ext[i]++;
           	l=i,r=i+ext[i]-1;
        }
        else if(Z[i-l+1]<r-i+1) //O(1)
            ext[i]=Z[i-l+1];
        else{
            ext[i]=r-i;
            while(1+ext[i]<=m&&i+ext[i]<=n&&str1[i+ext[i]]==str2[1+ext[i]])
                ext[i]++;
            l=i,r=i+ext[i]-1;
        }
    }
}
int main()
{
	
	scanf("%s%s",a+1,b+1);
	n=strlen(a+1),m=strlen(b+1);
	
	get_Z(m,b);//别传错了
	
	get_ext(n,a,m,b);
	
	ll suma=0,sumb=0;
	for(int i=1;i<=m;i++){
		suma^=(i*(Z[i]+1));
	}
	cout<<suma<<endl;
	
	for(int i=1;i<=n;i++){
		sumb^=(i*(ext[i]+1));
	}
	cout<<sumb<<endl;
	
	return 0;
}

Problem: CF1537E1 Erase and Extend (Easy Version and Hard Version)

给一个字符串s，有两种操作：1.删去最后一个字符。2.复制自身，s=s+s。
可以随意进行操作，也可以不进行操作，找到s进行操作侯获得得长度为k得字符串中字典序最小得字符串。

1<=n,k<=5000（easy 版本）
1<=n,k<=5e5  (hard 版本)

贪心得到所有可能得前缀，再比较O(n*k)

// Problem: CF1537E1 Erase and Extend (Easy Version)
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/CF1537E1
// Memory Limit: 250 MB
// Time Limit: 2000 ms

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <queue>
#include <set>
#include <map>
#include <vector>
#include <sstream>
#define pb push_back 
#define in insert
#define mem(f, x) memset(f,x,sizeof(f)) 
#define fo(i,a,n) for(int i=(a);i<=(n);++i)
#define fo_(i,a,n) for(int i=(a);i<(n);++i)
#define debug(x) cout<<#x<<":"<<x<<endl;
#define endl '\n'
using namespace std;

template<typename T>
ostream& operator<<(ostream& os,const vector<T>&v){for(int i=0,j=0;i<v.size();i++,j++)if(j>=5){j=0;puts("");}else os<<v[i]<<" ";return os;}
template<typename T>
ostream& operator<<(ostream& os,const set<T>&v){for(auto c:v)os<<c<<" ";return os;}

typedef pair<int,int>PII;
typedef pair<long,long>PLL;

typedef long long ll;
typedef unsigned long long ull; 
const int N=2e5+10,M=1e9+7;
ll n,m,_,k;

string get(string s,int k){
	while(s.size()<k){
		s+=s;
	}
	while(s.size()>k)
		s.pop_back();
	return s;
}

void solve()
{
	string s;cin>>n>>k>>s;
	string minn="",pre="";
	pre+=s[0];
	minn=get(pre,k);
	for(int i=1;i<n;i++){
		if(s[i]>s[0])break;
		pre+=s[i];
		minn=min(minn,get(pre,k));
	}
	cout<<minn<<endl;
}

int main()
{
	solve();
	return 0;
}

https://blog.csdn.net/weixin_43960414/article/details/118095536
给了两种很Nb的做法，一个是Z函数，一个是trick。
    发现字符串的规律还是很难啊！