字符串中的所有变位词

字符串中的所有变位词

题目:

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

解题思路: 思路与字符串中的变位词相似

双指针解法:

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        List<Integer> ans = new ArrayList();
        char[] cs = s.toCharArray(), cp = p.toCharArray();
        int lenS = cs.length, lenP = cp.length;
        if(lenS < lenP) {
            return ans;
        }

        int count[] = new int[26];
        for(char c : cp) {
            count[c - 'a']--;
        }

        int l = 0;
        for(int r = 0; r < lenS; r++) {
            count[cs[r] - 'a']++;
            while(count[cs[r] - 'a'] > 0) {
                count[cs[l] - 'a']--;
                l++;
            }
            if(r - l + 1 == lenP) {
                ans.add(l);
            }
        }

        return ans;
    } 
}

滑动窗口解法:

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        List<Integer> ans = new ArrayList();
        char[] cs = s.toCharArray(), cp = p.toCharArray();
        int lenS = cs.length, lenP = cp.length;
        if(lenS < lenP) {
            return ans;
        }

        int[] as = new int[26], ap = new int[26];
        for(int i = 0; i < lenP; i++) {
            as[cs[i] - 'a']++;
            ap[cp[i] - 'a']++;
        }

        if(Arrays.equals(as, ap)) {
            ans.add(0);
        }
        
        for(int i = lenP; i < lenS; i++) {
            as[cs[i] - 'a']++;
            as[cs[i - lenP] - 'a']--;
            if(Arrays.equals(as, ap)) {
                ans.add(i - lenP + 1);
            }
        }

        return ans;
    } 
}