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

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

滑动窗口+ 双指针

/**
 * 双指针，滑动窗口
 * @param s
 * @param p
 * @return
 */
public static List<Integer> findAnagrams(String s, String p) {
    List<Integer> ans = new ArrayList<>();
    int n = s.length(), m = p.length();
    // 分别记录一段范围内的字符出现的次数
    int[] c1 = new int[26], c2 = new int[26];
    for (int i = 0; i < m; i++) c2[p.charAt(i) - 'a']++;
    // l，r分别表示滑动窗口两端的下标
    for (int l = 0, r = 0; r < n; r++) {
        // 加入滑动窗口内
        c1[s.charAt(r) - 'a']++;
        // 移除滑动窗口的第一个元素
        if (r - l + 1 > m) c1[s.charAt(l++) - 'a']--;
        // 检测是否匹配
        if (check(c1, c2)) ans.add(l);
    }
    return ans;
}

static boolean check(int[] c1, int[] c2) {
    for (int i = 0; i < 26; i++) {
        if (c1[i] != c2[i]) return false;
    }
    return true;
}

测试用例

public static void main(String[] args) {
    String s = "cbaebabacd";
    String p = "abc";
    List<Integer> integerList = FindAnagrams.findAnagrams(s, p);
    System.out.println("FindAnagrams demo01 result : ");
    for (Integer integer : integerList) {
        System.out.print(" " + integer);
    }
    System.out.println("");

    s = "abab";
    p = "ab";
    integerList = FindAnagrams.findAnagrams(s, p);
    System.out.println("FindAnagrams demo02 result : ");
    for (Integer integer : integerList) {
        System.out.print(" " + integer);
    }
    System.out.println("");
}

测试结果

FindAnagrams demo01 result : 
 0 6
FindAnagrams demo02 result : 
 0 1 2