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

给定一个字符串 s 和一个非空字符串 p,找到 s 中所有是 p 的字母异位词的子串,返回这些子串的起始索引。 字符串只包含小写英文字母,并且字符串 s 和 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” 的字母异位词。

解题思路

滑动窗口

  1. class Solution {
  2. public:
  3. vector<int> findAnagrams(string s, string p) {
  4. unordered_map<char,int> need,window;
  5. for (char c : p) need[c]++;
  6. int left = 0,right = 0;
  7. int valid = 0;
  8. vector<int> res;
  9. while (right < s.size()) {
  10. char c = s[right++];
  11. if (need.count(c)) {
  12. window[c]++;
  13. if(window[c] == need[c]) valid++;
  14. }
  15. while(right - left >= p.size()) {
  16. if (valid == need.size())
  17. res.push_back(left);
  18. char d = s[left++];
  19. if (need.count(d)) {
  20. if(window[d] == need[d]) valid--;
  21. window[d]--;
  22. }
  23. }
  24. }
  25. return res;
  26. }
  27. };