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

题目描述:

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

解题思路:

根据p做一个滑动窗口，这道题就是看官方解法的，和字母异位词分组题目的解法类似，需要转化为ascii编码进行转换。

python代码:

class Solution:
    def findAnagrams(self, s: str, p: str) -> List[int]:
        #长度
        s_len,p_len = len(s),len(p)
        ans = []
        #26个字母长度
        s_count = [0]*26
        p_count = [0]*26
        #ascii编码的转化
        for i in range(p_len):
            s_count[ord(s[i])-97] +=1
            p_count[ord(p[i])-97] +=1
        if s_count == p_count:
            ans.append(0)
        #前面已经对比过p的长度了,这里只需要对比s-p
        for i in range(s_len-p_len):
            s_count[ord(s[i])-97] -=1
            #这里就是进行滑动窗口
            s_count[ord(s[i+p_len])-97] +=1
            if s_count == p_count:
                ans.append(i+1)
        return ans

java代码:

class Solution {
    public List<Integer> findAnagrams(String s, String p) {
        int s_len = s.length(), p_len = p.length() ;
        List<Integer> ans =new ArrayList<Integer>();
        int[] s_count = new int[26];
        int[] p_count = new int[26];
        for(int i=0;i<p_len;i++){
            ++s_count[s.charAt(i)- 'a'];
            ++p_count[p.charAt(i)-'a'];
        }
        if(Arrays.equals(s_count,p_count)){
            ans.add(0);
        }
        for(int i=0;i<s_len-p_len;i++){
            --s_count[s.charAt(i)-'a'];
            ++s_count[s.charAt(i+p_len)-'a'];
            if(Arrays.equals(s_count,p_count)){
                ans.add(i+1);
            }
        }
        return ans;


    }
}