LeetCode438 找到字符串中的所有字母异位数

题目

给定两个字符串 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 * 10⁴ 
 s 和 p 仅包含小写字母 

方法

滑动窗口法

设置长度为p的滑动窗口，依次对比滑动窗口的字符数和p中的字符数是否相同

• 时间复杂度：O(m+(n-m)*c),m为p字符串的长度，n为s的长度，m是初始化滑动窗口的时间，n-m是移动的时间，c是字符数组的长度，每次移动要判断c个字符是否相同
• 空间复杂度：O(c)

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        int sLen = s.length(),pLen = p.length();
        if(sLen<pLen){
            return new ArrayList<>();
        }
        int[] sNum = new int[26];
        int[] pNum = new int[26];
        List<Integer> ans = new ArrayList<>();
        //先把滑动窗口填进数据
        for(int i=0;i<pLen;i++) {
            pNum[p.charAt(i) - 'a']++;
            sNum[s.charAt(i) - 'a']++;
        }
        if(Arrays.equals(pNum,sNum)){
            ans.add(0);
        }
        for(int i=0;i<sLen-pLen;i++){
            sNum[s.charAt(i)-'a']--; //依次挪动滑动窗口
            sNum[s.charAt(i+pLen)-'a']++;
            if(Arrays.equals(pNum,sNum)){
                ans.add(i+1);
            }
        }
        return ans;
    }
}

优化的滑动窗口

diff表示不同的字母个数，如count[i]=2和count[i]=-1时diff都是1，除非变成0

• 时间复杂度：O(m+n+c),m为p字符串的长度，n为s的长度，m是初始化滑动窗口的时间，n-m是移动的时间，c是字符数组的长度，每次移动要判断c个字符是否相同
• 空间复杂度：O(c)

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        int sLen = s.length(),pLen = p.length();
        if(sLen<pLen) return new ArrayList<>();
        int[] count = new int[26];
        int diff = 0;
        List<Integer> ans = new ArrayList<>();
        for(int i=0;i<pLen;i++){
            count[p.charAt(i)-'a']--;
            count[s.charAt(i)-'a']++;
        }
        for(int i=0;i<26;i++){
            if(count[i]!=0){
                diff++;
            }
        }
        if(diff==0){
            ans.add(0);
        }
        for(int i=0;i<sLen-pLen;i++){
            if(count[s.charAt(i)-'a']==1){
                diff--;
            }else if(count[s.charAt(i)-'a']==0){
                diff++;
            }
            count[s.charAt(i)-'a']--;
            if(count[s.charAt(i+pLen)-'a']==-1){
                diff--;
            }else if(count[s.charAt(i+pLen)-'a']==0){
                diff++;
            }
            count[s.charAt(i+pLen)-'a']++;
            if(diff==0){
                ans.add(i+1);
            }
        }
        return ans;
    }
}