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

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

题目

给定两个字符串 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" 的异位词。

思路

• 可以使用滑动数组的概念
• 滑动数组的长度与p的长度相等，每向前一步，减掉原本数组中最前面的字符，新增正在遍历的字符，可以使用一个长度为26的数组来表示目前的滑动数组里各有多少个字符，然后与p（也生成一个长度为26的数组）相比，如果相同，就加入到结果集中。

实现代码

public List<Integer> findAnagrams(String s, String p) {
    if (s.length() < p.length()) return Collections.emptyList();
    List<Integer> result = new ArrayList<>();
    int[] slist = new int[26];
    int[] plist = new int[26];
    for (int i = 0; i < p.length(); i++) {
        slist[s.charAt(i) - 'a']++;
        plist[p.charAt(i) - 'a']++;
    }
    if (Arrays.equals(slist, plist)){
        result.add(0);
    }
    for (int i = 0; i < s.length() - p.length(); i++) {
        slist[s.charAt(i) - 'a']--;
        slist[s.charAt(i + p.length()) - 'a']++;
        if (Arrays.equals(slist, plist)){
            result.add(i + 1);
        }
    }
    return result;
}