leetcode 3008. 找出数组中的美丽下标 II

题目如下

数据范围

复制代码
首先使用kmp算法统计两个串的出现下标,随后使用双指针匹配就行。这道题纯纯kmp模板题。

通过代码

cpp 复制代码
class Solution {
public:
    vector<int> beautifulIndices(string s, string a, string b, int k) {
        int al = a.size();
        int bl = b.size();
        int n = s.size();
        vector<int> af;
        vector<int> bf;
        vector<int> ans;
        vector<int> an(al, 0), bn(bl, 0);
        for (int i = 1, j = 0; i < al; i++) {
            while (j > 0 && a[i] != a[j])
                j = an[j - 1];
            if (a[i] == a[j])
                j++;
            an[i] = j;
        }
        for (int i = 1, j = 0; i < bl; i++) {
            while (j > 0 && b[i] != b[j])
                j = bn[j - 1];
            if (b[i] == b[j])
                j++;
            bn[i] = j;
        }
        int j = 0;
        int i = 0;
        while (i < n) {
            if (s[i] == a[j]) {
                i++;
                j++;
            } else {
                if (j > 0) {
                    j = an[j - 1];
                } else {
                    i++;
                }
            }
            if (al == j) {
                //   cout << i<< endl;
                af.push_back(i - al);
                j = an[j - 1];
            }
        }
        i = j = 0;
        while (i < n) {
            if (s[i] == b[j]) {
                i++;
                j++;
            } else {
                if (j > 0) {
                    j = bn[j - 1];
                } else {
                    i++;
                }
            }
            if (bl == j) {
                bf.push_back(i - bl);
                j = bn[j - 1];
            }
        }
        int l = 0, r = 0;
        while (l < af.size() && r < bf.size()) {
            if (abs(af[l] - bf[r]) <= k) {
                ans.push_back(af[l]);
                l++;
            } else if (af[l] < bf[r]) {
                l++;
            } else {
                r++;
            }
        }
        return ans;
    }
};
相关推荐
贾全7 分钟前
第十章:HIL-SERL 真实机器人训练实战
人工智能·深度学习·算法·机器学习·机器人
GIS小天22 分钟前
AI+预测3D新模型百十个定位预测+胆码预测+去和尾2025年7月4日第128弹
人工智能·算法·机器学习·彩票
oioihoii35 分钟前
C++11 forward_list 从基础到精通:原理、实践与性能优化
c++·性能优化·list
满分观察网友z39 分钟前
开发者的“右”眼:一个树问题如何拯救我的UI设计(199. 二叉树的右视图)
算法
m0_6873998443 分钟前
写一个Ununtu C++ 程序,调用ffmpeg API, 来判断一个数字电影的视频文件mxf 是不是Jpeg2000?
开发语言·c++·ffmpeg
森焱森2 小时前
无人机三轴稳定化控制(1)____飞机的稳定控制逻辑
c语言·单片机·算法·无人机
循环过三天2 小时前
3-1 PID算法改进(积分部分)
笔记·stm32·单片机·学习·算法·pid
呆瑜nuage2 小时前
数据结构——堆
数据结构
Ronin3052 小时前
【C++】类型转换
开发语言·c++
蓝澈11212 小时前
弗洛伊德(Floyd)算法-各个顶点之间的最短路径问题
java·数据结构·动态规划