面试算法-142-找到字符串中所有字母异位词

题目

给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。

异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。

示例 1:

输入: s = “cbaebabacd”, p = “abc”
输出: [0,6]
解释:
起始索引等于 0 的子串是 “cba”, 它是 “abc” 的异位词。
起始索引等于 6 的子串是 “bac”, 它是 “abc” 的异位词。

解

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        List<Integer> result = new ArrayList<>();
        if(s.length() < p.length()){
            return result;
        }
        
        int n = p.length();
        int[] letter = new int[26];
        for (int i = 0; i < n; i++) {
            char sCh = s.charAt(i);
            char pCh = p.charAt(i);
            letter[sCh - 'a']--;
            letter[pCh - 'a']++;
        }
        
        int i = n;
        for (; i < s.length(); i++) {
            if (allZero(letter)) {
                result.add(i - n);
            }
            char rightCh = s.charAt(i);
            char leftCh = s.charAt(i - n);
            letter[rightCh - 'a']--;
            letter[leftCh - 'a']++;
        }
        if (allZero(letter)) {
            result.add(i - n);
        }
        return result;
    }

    public boolean allZero(int[] letter) {
        int n = letter.length;
        for (int i = 0; i < n; i++) {
            if (letter[i] != 0) {
                return false;
            }
        }
        return true;
    }
}