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

一、题目描述

二、示例

三、难度

中等

四、代码

Java版

方法：滑动窗口 + 数组

public class Solution {
    public static List<Integer> findAnagrams(String s, String p) {
        int len1 = s.length();
        int len2 = p.length();
        List<Integer> list = new ArrayList<>();
        if (len1 < len2) return list;
        /*
            1.记录s1和s2每个字母的个数
            2.定义一个长度为26的数组，下标0-25代表26个字母
            3.数组比较直接用 equals
         */
        int[] arr_s = new int[26];
        int[] arr_p = new int[26];

        for (int i = 0; i < len2; ++i) {
            arr_s[s.charAt(i) - 'a']++;
            arr_p[p.charAt(i) - 'a']++;
        }
        if (Arrays.equals(arr_p, arr_s)) list.add(0);
        // 滑动窗口  i相当于right右指针   i - len2相当于left左指针
        for (int i = len2; i < len1; ++i) {
            // 窗口移入的元素
            arr_s[s.charAt(i) - 'a']++;
            // 窗口移出的元素
            arr_s[s.charAt(i - len2) - 'a']--;
            if (Arrays.equals(arr_p, arr_s)) list.add(i-len2+1);
        }
        return list;
    }

    public static void main(String[] args) {
        int[] arr = new int[]{-1,0};
        //System.out.println(removeElement(new int[]{0,1,2,2,3,0,4,2},2));

        System.out.println(findAnagrams("cbaebabacd","abc"));
        System.out.println(findAnagrams("abab","ab"));
        System.out.println(findAnagrams("a","a"));
    }
}

Python版

class Solution(object):

    def findAnagrams(self, s, p):
        """
        :type s: str
        :type p: str
        :rtype: List[int]
        """
        s_len, p_len = len(s), len(p)
        if (s_len < p_len): return []

        res = []
        s_arr, p_arr = [0]*26, [0]*26
        for i in range(p_len):
            s_arr[ord(s[i]) - ord('a')] += 1
            p_arr[ord(p[i]) - ord('a')] += 1
        if s_arr == p_arr: res.append(0)

        for i in range(p_len, s_len):
            # 新移入窗口的元素
            s_arr[ord(s[i]) - ord('a')] += 1
            # 移出窗口的元素
            s_arr[ord(s[i - p_len]) - ord('a')] -= 1
            if s_arr == p_arr: res.append(i - p_len + 1)

        return res


if __name__ == '__main__':
    print(Solution().findAnagrams('abab','ab'))