LeetCode438. 找到字符串中所有字母异位词 （字符串--滑动窗口）

https://leetcode-cn.com/problems/find-all-anagrams-in-a-string/submissions/

给定一个字符串 s 和一个非空字符串 p，找到 s 中所有是 p 的字母异位词的子串，返回这些子串的起始索引。
字符串只包含小写英文字母，并且字符串 s 和 p 的长度都不超过 20100。
说明：
字母异位词指字母相同，但排列不同的字符串。
不考虑答案输出的顺序。
示例 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” 的字母异位词。

思路：见代码

/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
/* 思路：
1、构建p_arr[26],并赋值
2、构建结果集
3、遍历s,建立一个滑动窗口，寻找满足答案序列
*/
bool check_all_zero(int *arr){
    int i = 0;
    for(; i < 26; i++){
        if(0 != arr[i]){
            return false;
        }
    }
    return true;
}

int* findAnagrams(char* s, char* p, int* returnSize) {
    int plen = strlen(p);
    int slen = strlen(s);
    if(NULL == s || NULL == p || slen < plen){
        return NULL;
    }
    // step 1
    int p_arr[26] = {0};
    while(*p != '\0'){
        p_arr[*p - 'a']++;
        p++;
    }
    
    // step 2
    int *ret = (int *)malloc(slen * sizeof(int));
    memset(ret, -1, slen * sizeof(int));
    
    // step 3
    int i = 0;
    char *head = s;
    for(; i < plen; i++){
        p_arr[*s - 'a']--;
        s++;
    }
    *returnSize = 0;
    if(true == check_all_zero(p_arr)){
        ret[*returnSize] = 0;
        (*returnSize)++;
    }
    
    for(i = 1; i <= slen - plen; i++){
        p_arr[*head - 'a']++;   // 前面的加上
        p_arr[*s - 'a']--;      // 后面的补上
        head++;
        s++;
        //check
        if(true == check_all_zero(p_arr)){
            ret[*returnSize] = i;
            (*returnSize)++;
        }
    }
    return ret;
}