LeetCode 3036. 匹配模式数组的子数组数目 II（KMP算法、字符串匹配）

岁忧

于 2024-07-12 17:36:31 发布

阅读量102

点赞数 6

分类专栏： LeetCode 文章标签：算法 leetcode 职场和发展 c++ 数据结构字符串匹配 KMP算法

本文链接：https://blog.csdn.net/weixin_46028214/article/details/140385067

版权

LeetCode 专栏收录该内容

356 篇文章 0 订阅

订阅专栏

题目：3036. 匹配模式数组的子数组数目 II

在这里插入图片描述

思路：先处理出nums数据，将其变为含有-1、0、1的数组。再使用KMP算法。(KMP模板题) acwing 831. KMP字符串，时间复杂度0(n+m)

class Solution {
public:
    int countMatchingSubarrays(vector<int>& nums, vector<int>& pattern) {
        int n=nums.size(),m=pattern.size();
        for(int i=0;i+1<n;i++){
            if(nums[i]<nums[i+1]) nums[i]=1;
            else if(nums[i]==nums[i+1]) nums[i]=0;
            else nums[i]=-1;
        }
        vector<int> p(m,-1);
        for(int i=1,j=-1;i<m;i++){
            while(j!=-1&&pattern[j+1]!=pattern[i]) j=p[j];
            if(pattern[j+1]==pattern[i]) j++;
            p[i]=j;
        }
        int ans=0;
        for(int i=0,j=-1;i<n;i++){
            while(j!=-1&&pattern[j+1]!=nums[i]) j=p[j];
            if(pattern[j+1]==nums[i]) j++;
            if(j==m-1){
                ans++;
                j=p[j];
            }
        }
        return ans;
    }
};