LeetCode438. 找到字符串中所有字母异位词Golang版

LeetCode438. 找到字符串中所有字母异位词Golang版

1. 问题描述

给定一个字符串 s 和一个非空字符串 p，找到 s 中所有是 p 的字母异位词的子串，返回这些子串的起始索引。

字符串只包含小写英文字母，并且字符串 s 和 p 的长度都不超过 20100。

说明：

字母异位词指字母相同，但排列不同的字符串。
不考虑答案输出的顺序。

2. 思路

滑动窗口

3. 代码

3.1. 一种写法

func findAnagrams(s string, p string) []int {
    var res []int
    var freq [256]int
    if len(s) < len(p) || len(s) == 0 {
        return res
    }

    for i := 0; i < len(p); i++ {
        freq[p[i] - 'a']++
    }

    left, right := 0, 0
    count := len(p)
    for right < len(s) {
        // fmt.Println(right)
        if freq[s[right] - 'a'] > 0 {
            count--
        }
       
        if count == 0 {
            res = append(res, left)
        }

        freq[s[right] - 'a']--
        right++

        if right - left == len(p) {
            if freq[s[left] - 'a'] >= 0 {
                count++
            }
            freq[s[left] - 'a']++
            left++
        }
    }
    return res
}

第二种写法

func findAnagrams(s string, p string) []int {
    var res []int
    if len(s) < len(p) {
        return res
    }

    var pCnt [26]int
    var sCnt [26]int

    for i := 0; i < len(p); i++ {
        pCnt[p[i] - 'a']++
    } 

    left := 0
    for right := 0; right < len(s); right++ {
        sCnt[s[right] - 'a']++
        for sCnt[s[right] - 'a'] > pCnt[s[right] - 'a'] {
            sCnt[s[left] - 'a']--
            left++
        }
        if right - left + 1 == len(p) {
            res = append(res, left)
        }
    }
    return res
}