保研机试训练Day-06

单调队列 && KMP

Posted by Jeffzzc on March 25, 2025

大纲

单调队列

#include<iostream>
using namespace std;

int n,k;
const int N = 1000010;
int a[N],q[N];
int hh=0,tt=-1;
int main(){
    scanf("%d%d",&n,&k);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
        if(i-k+1 > q[hh]) hh++;
        while(hh <= tt && a[i] <= a[q[tt]]) tt--;
        q[++tt] = i;
        if(i+1>=k) printf("%d ",a[q[hh]]);
    }
    printf("\n");
    hh=0,tt=-1;
    for(int i=0;i<n;i++){
        if(i-k+1 > q[hh]) hh++;
        while(hh <= tt && a[i] >= a[q[tt]]) tt--;
        q[++tt] = i;
        if(i+1>=k) printf("%d ",a[q[hh]]);
    }
    return 0;
}

KMP

给定一个字符串 S,以及一个模式串 P,所有字符串中只包含大小写英文字母以及阿拉伯数字。

模式串 P 在字符串 S 中多次作为子串出现。

求出模式串 P 在字符串 S 中所有出现的位置的起始下标。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10,M=1e6+10;
int n,m;
char p[N],s[M];
int ne[N];

int main(){
    // p s下标为1
    cin>>n>>p+1>>m>>s+1;
    // 求next过程
    for(int i=2,j=0;i<=n;i++){
        while(j && p[i] != p[j+1]) j=ne[j];
        if(p[i] == p[j+1]) j++;
        ne[i] = j;
    }

    // 匹配过程
    for(int i=1,j=0;i<=m;i++){
        while(j && s[i]!=p[j+1]){
            j = ne[j];
        }
        if(s[i] == p[j+1]) j++;
        if(j==n){
            // 成功
            printf("%d ",i-n);
        }
    }
    return 0;
}