LeetCode 438. 找到字符串中所有字母异位词

题目描述

给定一个字符串 和一个非空字符串 p,找到 中所有是 的字母异位词的子串,返回这些子串的起始索引。

字符串只包含小写英文字母,并且字符串 和 的长度都不超过 20100。

说明:

  • 字母异位词指字母相同,但排列不同的字符串。
  • 不考虑答案输出的顺序。

示例 1:

输入:
s: "cbaebabacd" p: "abc"

输出:
[0, 6]

解释:
起始索引等于 0 的子串是 "cba", 它是 "abc" 的字母异位词。
起始索引等于 6 的子串是 "bac", 它是 "abc" 的字母异位词。

 示例 2:

输入:
s: "abab" p: "ab"

输出:
[0, 1, 2]

解释:
起始索引等于 0 的子串是 "ab", 它是 "ab" 的字母异位词。
起始索引等于 1 的子串是 "ba", 它是 "ab" 的字母异位词。
起始索引等于 2 的子串是 "ab", 它是 "ab" 的字母异位词。

问题分析

此题用滑动窗口解决。先判断如果字符串s的长度小于字符串p的长度,那么返回{}。然后创建保存答案的数组ans,创建保存字符串s中字符出现次数的数组preq_s,大小为26,初始值为0。创建保存字符串p中字符出现次数的数组preq_p,大小为26,初始值为0。然后for循环p.size()次,把字符串s的前p.size()个字符的出现次数保存在数组preq_s中,把字符串p的前p.size()个字符的出现次数保存在数组preq_p中。然后判断如果数组preq_s和数组preq_p相同,则将0加入ans中。接下来开始滑动窗口,先定义窗口的左边界为 l ,初始值为0,定义窗口的右边界的下一个位置为 r ,初始值为p.size()。如果 r 在字符串s内,那么将窗口向右移动一个单位:将 r 位置的字符s[r]的出现次数在数组preq_s中加1,然后将 l 位置的字符s[l]的出现次数在数组preq_s中减1。然后将 r 和 l 都加1。然后判断如果数组preq_s和数组preq_p相同,则将 l 加入ans中。如此进行,当滑动窗口的右边界的下一个位置 r不在字符串s中时,这时滑动窗口已经顶到了字符串s的右边界,所以退出while循环,返回ans。

代码实现

class Solution {
public:
    vector<int> findAnagrams(string s, string p) {
        if(s.size() < p.size())
            return {};
        vector<int> ans;
        vector<int> preq_s(26, 0), preq_p(26, 0);
        for(int i = 0; i < p.size(); i++){
            ++preq_s[s[i] - 'a'];
            ++preq_p[p[i] - 'a'];
        }
        if(preq_s == preq_p)
            ans.push_back(0);
        int l = 0;
        int r = p.size();
        while(r < s.size()){
            ++preq_s[s[r++] - 'a'];
            --preq_s[s[l++] - 'a'];
            if(preq_s == preq_p)
                ans.push_back(l);
        }
        return ans;
    }
};

猜你喜欢

转载自blog.csdn.net/mengyujia1234/article/details/89694771