[菜鸟训练]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” 的字母异位词。

解题思路：

滑动窗口+数组。 使用数组来存储字符串p中出现的字符的种类和个数，方便在操作滑动数组时与p进行字母异位词的比较。详细见代码。

代码：

public class LC438 {
    //滑动窗口
    //时间复杂度O(n) 空间复杂度O(1)
    public List<Integer> findAnagrams(String s, String p) {
        //记录p中字符出现的元素和个数，方便之后寻找字母异位词的时候进行比较
        char[] need = new char[26];
        for (int i = 0; i < p.length(); i++){
            need[p.charAt(i) - 'a']++;
        }
        //滑动窗口过程中的中间数组
        char[] window = new char[26];
        List<Integer> ansList = new ArrayList<>();
        //滑动窗口的前端和后端
        int start = 0, end = 0;
        while (end < s.length()){
            //将数据加入窗口
            window[s.charAt(end) - 'a']++;
            //窗口中的元素已经和目标字符串中的元素数目相等
            if ((end - start + 1) == p.length()){
                if (Arrays.equals(window,need) == true){
                    ansList.add(start);
                }
                //不管匹不匹配都将滑动窗口后移
                window[s.charAt(start) - 'a']--;
                start++;
            }
            end++;
        }
        return ansList;
    }

    public static void main(String[] args) {
        LC438 obj = new LC438();
        System.out.println(obj.findAnagrams("cbaebabacd", "abc"));
    }
}