LeetCode hot100-9

题目描述

题目链接：找到字符串中所有字母异位词

给定两个字符串 s 和 p，找到 s中所有 p的 异位词的子串，返回这些子串的起始索引。不考虑答案输出的顺序。

示例 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" 的异位词。

提示:

• 1 <= s.length, p.length <= 3 * 104
• s 和 p 仅包含小写字母

思路解析

判断异位词思路：因为字符串中只有小写字母，所以利用一个大小为26的数组进行统计每一个字母的数量，当每个字母数量相同时则为异位词。

滑动窗口进行遍历每一个长度跟p相同的字串：左端点记录字串开始位置，是异位词时向答案数组中添加下标，右端点记录字串最后一个元素下一个元素的位置，滑动窗口向后移的时候该位置元素计数+1，左端点位置元素计数-1。

注意：不管先判断异位词还是先滑动窗口，不要漏掉判断第一个和最后一个字串。

代码实现

class Solution {
public:
    vector<int> findAnagrams(string s, string p) {
        if(s.size()<p.size())return {};//如果s长度小于p则不可能有子串为p的异位词
        vector<int>vec;//答案数组
        int n=p.size();
        int cnt[26];//计数数组
        for(int i=0;i<n;i++){//初始化
            cnt[p[i]-'a']++;
            cnt[s[i]-'a']--;
        }
        for(int l=0,r=n;r<=s.size();l++,r++){//滑动窗口更新计数数组判断是否为异位词
            int k=1;
            for(auto it:cnt)//遍历计数数组，全为0时表示为异位词
                if(it!=0){
                    k=0;
                    break;
                }
            if(k)vec.push_back(l);
            if(r==s.size())break;
            cnt[s[l]-'a']++;
            cnt[s[r]-'a']--;
        }
        return vec;
    }
};