LeetCode/滑动窗口/2. 找到字符串中所有字母异位词.md

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

题目

给定两个字符串 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 仅包含小写字母

思路

• 维护滑动窗口
  • 大小等于目标异位词字符串长度
  • 注意特判
• 相等判定
  • Arrays.equals(src, tar)

代码

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        // Special
        if (s.length() < p.length()) return new ArrayList<Integer>();
        // Init
        int[] tar = new int[26];
        int[] src = new int[26];
        List<Integer> ans = new ArrayList<>();

        // Init Window
        for (int index = 0; index < p.length(); index++) {
            tar[p.charAt(index) - 'a']++;
            src[s.charAt(index) - 'a']++;
        }

        if (Arrays.equals(tar, src)) {
            ans.add(0);
        }
        // Traverse: Window [l, r]
        for (int l = 1, r = p.length(); r < s.length(); l++, r++) {
            // Update
            src[s.charAt(l - 1) - 'a']--;
            src[s.charAt(r) - 'a']++;
            if (Arrays.equals(tar, src)) {
                ans.add(l);
            }
        }

        return ans;
    }
}