kmp
kmp 是模式串匹配的算法,本来最坏时间复杂度可以达到 $\operatorname{O}(n\times m)$,但是 kmp 可以将复杂度优化到 $\operatorname{O}(n+m)$。
kmp 有个很重要的东西,叫做 $nxt$ 失配数组。比如对于一个字符串 $s$,它的失配数组 $nxt_n$ 就是 $s$ 的最大前缀等于后缀的长度。$\operatorname{O}(n\times m)$ 的算法的劣势在于每一次失配都要重头,但是 kmp 是从上一次最大失配数组 $nxt_{nxt_n}$ 开始匹配的,这样会优化很多无意义匹配的时间。
设定一个 $pos$ 为匹配开始的位置,然后通过 $nxt$ 数组来跳,可以达到 $\operatorname{O}(n+m)$ 的时间复杂度。
int len1=strlen(s1+1);
int len2=strlen(s2+1);
int pos=0;
for(int i=2;i<=len2;++i){
while(pos&&s2[i]!=s2[pos+1]){
pos=nxt[pos];//跳 nxt
}
if(s2[i]==s2[pos+1]){
++pos;//匹配成功
}
nxt[i]=pos;//存入 nxt
}
pos=0;
for(int i=1;i<=len1;++i){
while(pos&&s1[i]!=s2[pos+1]){
pos=nxt[pos];//跳 nxt
}
if(s1[i]==s2[pos+1]){
++pos;//匹配成功
}
if(pos==len2){
printf("%d %d\n",i,i+len2-1);//一个位置
}
}
例题
首先求出 nxt 数组,然后在 nxt 数组上 dp,设 $dp_i$ 表示前缀 $i$ 的答案,只可能从 $dp_{nxt_i}$ 和 $dp_i$ 转移过来,具体开桶实现。
#include<bits/stdc++.h>
#define MAXN 500005
using namespace std;
char s[MAXN];
int nxt[MAXN],dp[MAXN],mp[MAXN];
int main(){
scanf("%s",s+1);
int len=strlen(s+1);
int pos=0;
for(int i=2;i<=len;++i){
while(pos&&s[i]!=s[pos+1]){
pos=nxt[pos];
}
if(s[i]==s[pos+1]){
++pos;
}
nxt[i]=pos;
}
for(int i=1;i<=len;++i){
dp[i]=i;
if(mp[dp[nxt[i]]]>=i-nxt[i]){
dp[i]=dp[nxt[i]];
}
mp[dp[i]]=i;
}
printf("%d",dp[len]);
return 0;
}
字符串哈希
字符串哈希的思想是,通过把字符串看做 $k$ 进制数来进行存储和比较。
- 优点:相较于直接字符比较,更加迅速,而且能够 $\operatorname{O}(1)$ 查询某段子区间的哈希值。
- 缺点:容易冲突。
为了应对冲突,我们需要对哈希进制做一些优化,模数也需要非常极品。下面,给出预处理模版代码。
typedef unsigned long long ull;
ull base[MAXN],pre[MAXN],suf[MAXN];//进制、前缀哈希、后缀哈希
int len;
char s[MAXN];
inline ull get_pre(int l,int r){//O(1) 查询子区间哈希值
return ((pre[r]-pre[l-1]*base[r-l+1])%MOD+MOD)%MOD;
}
inline ull get_suf(int l,int r){//O(1) 查询子区间哈希值
return ((suf[l]-suf[r+1]*base[r-l+1])%MOD+MOD)%MOD;
}
inline void prework(){
base[0]=1;
for(int i=1;i<MAXN;++i){
base[i]=(base[i-1]*HASH)%MOD;
}
for(int i=1;i<=len;++i){
pre[i]=(pre[i-1]*HASH+s[i]+DIF)%MOD;//加上偏移量防卡
}
for(int i=len;i>=1;--i){
suf[i]=(suf[i+1]*HASH+s[i]+DIF)%MOD;//加上偏移量防卡
}
}
有时候,可以采取双模哈希来进行防卡,这样被卡的几率很小。
typedef unsigned long long ull;
ull base1[MAXN],pre1[MAXN],suf1[MAXN];
ull base2[MAXN],pre2[MAXN],suf2[MAXN];//进制、前缀哈希、后缀哈希
int len;
char s[MAXN];
inline ull get_pre1(int l,int r){
return ((pre1[r]-pre1[l-1]*base1[r-l+1])%MOD1+MOD1)%MOD1;
}
inline ull get_suf1(int l,int r){
return ((suf1[l]-suf1[r+1]*base1[r-l+1])%MOD1+MOD1)%MOD1;
}
inline ull get_pre2(int l,int r){//O(1) 查询子区间哈希值
return ((pre2[r]-pre2[l-1]*base2[r-l+1])%MOD2+MOD2)%MOD2;
}
inline ull get_suf2(int l,int r){
return ((suf2[l]-suf2[r+1]*base2[r-l+1])%MOD2+MOD2)%MOD2;
}
inline void prework(){
base1[0]=base2[0]=1;
for(int i=1;i<MAXN;++i){
base1[i]=(base1[i-1]*HASH1)%MOD1;
base2[i]=(base2[i-1]*HASH2)%MOD2;
}
for(int i=1;i<=len;++i){
pre1[i]=(pre1[i-1]*HASH1+s[i]+DIF1)%MOD1;
pre2[i]=(pre2[i-1]*HASH1+s[i]+DIF2)%MOD2;//加上偏移量防卡
}
for(int i=len;i>=1;--i){
suf1[i]=(suf1[i+1]*HASH1+s[i]+DIF1)%MOD1;
suf2[i]=(suf2[i+1]*HASH2+s[i]+DIF2)%MOD2;//加上偏移量防卡
}
}
例题
这一道题目可以先把哈希值处理出来,然后发现可以枚举中间点,然后向左右二分。由于向左延伸 $n$ 格是回文串,那么 $n-1$ 格肯定也是回文串。所以满足单调性可以二分。由于本题有 Hack 数据卡自然溢,所以要打双模。
#include<bits/stdc++.h>
#define MAXN 500005
#define HASH1 31
#define HASH2 29
#define MOD1 193910017
#define MOD2 1000000009
#define ADD 131
using namespace std;
typedef long long ull;
int len;
char s[MAXN];
ull base1[MAXN],pre1[MAXN],suf1[MAXN];
ull base2[MAXN],pre2[MAXN],suf2[MAXN];
inline ull get_pre1(int l,int r){
if(!l){
return 0;
}
return ((pre1[r]-pre1[l-1]*base1[r-l+1])%MOD1+MOD1)%MOD1;
}
inline ull get_suf1(int l,int r){
if(!l){
return 0;
}
return ((suf1[l]-suf1[r+1]*base1[r-l+1])%MOD1+MOD1)%MOD1;
}
inline ull get_pre2(int l,int r){
if(!l){
return 0;
}
return ((pre2[r]-pre2[l-1]*base2[r-l+1])%MOD2+MOD2)%MOD2;
}
inline ull get_suf2(int l,int r){
if(!l){
return 0;
}
return ((suf2[l]-suf2[r+1]*base2[r-l+1])%MOD2+MOD2)%MOD2;
}
int main(){
scanf("%d %s",&len,s+1);
base1[0]=base2[0]=1;
for(int i=1;i<=len;++i){
base1[i]=(base1[i-1]*HASH1)%MOD1;
base2[i]=(base2[i-1]*HASH2)%MOD2;
pre1[i]=(pre1[i-1]*HASH1+(s[i]-'0'+ADD))%MOD1;
pre2[i]=(pre2[i-1]*HASH2+(s[i]-'0'+ADD))%MOD2;
}
for(int i=len;i>=1;--i){
suf1[i]=(suf1[i+1]*HASH1+('1'-s[i]+ADD))%MOD1;
suf2[i]=(suf2[i+1]*HASH2+('1'-s[i]+ADD))%MOD2;
}
ull ans=0;
for(int i=1;i<len;++i){
int l=0,r=min(i,len-i),res=0;
if(s[i]==s[i+1]){
continue;
}
while(l<=r){
int mid=(l+r)>>1;
if(get_pre1(i-mid,i)==get_suf1(i+1,i+1+mid)&&get_pre2(i-mid,i)==get_suf2(i+1,i+1+mid)){
l=mid+1;
res=l;
}else{
r=mid-1;
}
}
ans+=res;
}
printf("%llu",ans);
return 0;
}
字典树
字典树是一种 $k$ 叉树结构。原理是每一个节点都有一个布尔值 $flag$,判断是否是一个字符串的结尾。每一条边都有一个字符,表示前面的字符拼接起来就是字符串。这种数据结构能够 $\operatorname{O}(n)$ 查找前缀。
int cnt,trie[MAXN][MAXT],flag[MAXN];
inline int turn(char c);//字符映射函数
inline void insert(string s){
int root=0;
for(int i=0;i<s.size();++i){
int p=turn(s[i]);
if(trie[root][p]){//有没有节点创建过
root=trie[root][p];//有就跳
}else{
root=trie[root][p]=++cnt;//没有就建边
}
}
flag[root]=true;//标记末尾
}
inline bool find(string s){
int root=0;
for(int i=0;i<s.size();++i){
int p=turn(s[i]);
if(!trie[root][p]){//如果没有,直接返回
return false;
}
root=trie[root][p];//跳
}
return flag[root];//有没有这个末尾
}
例题
这一道题目考虑贪心证明。如果在某一层,$u$ 需要比 $v$ 先,那么就建一条边。如果在下一层,出现了需要 $v$ 比 $u$ 先的情况,那就冲突了,不可能是最优。也就是判环或者用种类并查集。用 Topu 或者 Tarjan 都可以判环。
#include<bits/stdc++.h>
#define MAXN 30003
#define MAXM 26
#define MAXK MAXN*10
using namespace std;
struct node{
int nxt[MAXM];
bool end;
}tree[MAXK];
vector<string> ans;
int n,cnt,indeg[MAXM];
bool vis[MAXM][MAXM];
string s[MAXN];
inline void insert(string str){
int root=0;
for(int i=0;i<str.size();++i){
int dot=str[i]-'a';
if(!tree[root].nxt[dot]){
tree[root].nxt[dot]=++cnt;
}
root=tree[root].nxt[dot];
}
tree[root].end=true;
}
inline bool addedge(string str){
int root=0;
for(int i=0;i<str.size();++i){
int dot=str[i]-'a';
if(tree[root].end){
return false;
}
for(int j=0;j<MAXM;++j){
if(tree[root].nxt[j]&&dot!=j&&!vis[dot][j]){
++indeg[j];
vis[dot][j]=true;
}
}
root=tree[root].nxt[dot];
}
return true;
}
inline bool topusort(){
queue<int> q;
for(int i=0;i<MAXM;++i){
if(!indeg[i]){
q.push(i);
}
}
while(!q.empty()){
int front=q.front();
q.pop();
for(int i=0;i<MAXM;++i){
if(vis[front][i]){
--indeg[i];
if(!indeg[i]){
q.push(i);
}
}
}
}
for(int i=0;i<MAXM;++i){
if(indeg[i]){
return false;
}
}
return true;
}
int main(){
ios::sync_with_stdio(false);
int n;
cin>>n;
for(int i=1;i<=n;++i){
cin>>s[i];
insert(s[i]);
}
for(int i=1;i<=n;++i){
memset(indeg,0,sizeof(indeg));
memset(vis,0,sizeof(vis));
if(addedge(s[i])&&topusort()){
ans.push_back(s[i]);
}
}
cout<<ans.size();
for(int i=0;i<ans.size();++i){
cout<<endl<<ans[i];
}
return 0;
}
标签:指南,MOD1,MOD2,return,int,MAXN,ull,字符串
From: https://www.cnblogs.com/hisy/p/18421160