【C语言刷LeetCode】525. 连续数组（M)

【

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

示例 1:

输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。
示例 2:

输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。
 

提示：

1 <= nums.length <= 105
nums[i] 不是 0 就是 1

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/contiguous-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

】

这道题首先想到的是滑动窗口，但是做起来超时了。

然后看了题解，一个重要思想，把0改成-1，那么相同数量的0和1的和就变成-1和1的和，为0。理所当然想到了前缀和。

然后是哈希，用数组模拟哈希，但初始化又遇到点困难，0该如何处理。首先哈希数组的初始化值不能为0，然后还要加个基10000，再到现在我也不太明白当初为何这么写，因为我已经学会了uthash,之前方法如下：

int findMaxLength(int* nums, int numsSize){
    int zeroCnt = 0;
    int oneCnt = 0;
    int i, j; // 这题首先想到的是滑动窗口
    int preSum[numsSize];
    int ret = 0;
    int hash[204800];
    int base = 100000;
 
    memset(hash, 0x3f, sizeof(hash));
 
    preSum[0] = nums[0] ? 1 : -1;
    hash[preSum[0] + base] = 0;
 
    for (i = 1; i < numsSize; i++) { // 先搞个前缀和, 把 0 改成 -1
        if (nums[i] == 1) {
            preSum[i] = preSum[i - 1] + 1;
        } else {
            preSum[i] = preSum[i - 1] - 1;
        }
        
        if (hash[preSum[i] + base] != 0x3f3f3f3f) {
            ret = fmax(ret, i - hash[preSum[i] + base]);
        } else {
            hash[preSum[i] + base] = i;
        }
    }
 
    for (i = numsSize - 1; i >= 0; i--) {
        if (preSum[i] == 0) { // 如果为0，可以计算ret了
            ret = fmax(ret, i + 1);
            break;
        }
    }
 
    return ret;
}

UThash:

突然发现用了uthash，题的难度降了一个等级。一样把0改-1；

然后把前缀和存到hash里，如果后面又相同的值的话，证明中间的字段是满足题意的，思路非常简单，代码如下：

* 方法二，用uthash
主要用到 HASH_ADD_INT 和 HASH_FIND_INT 两个函数
 
*/
 
typedef struct { // UT hash标准模板
    int key; // 哈希的ID
    int val; // 哈希存的值，这道题存的是cnt
    UT_hash_handle hh; // 这行固定写法；
} UT_HASH;
 
int findMaxLength(int* nums, int numsSize){
    UT_HASH *hashTable = NULL;
    UT_HASH *tmp = malloc(sizeof(UT_HASH));
    int ret = 0;
    int cnt = 0;
 
    tmp->key = 0;
    tmp->val = -1;
 
    HASH_ADD_INT(hashTable, key, tmp);
 
    for (int i = 0; i < numsSize; i++) {
        if (nums[i] == 1) {
            cnt++;
        } else {  // 如果是0的话，减1处理
            cnt--;
        }
 
        HASH_FIND_INT(hashTable, &cnt, tmp);
        if (tmp != NULL) {
            int preval = tmp->val;
            ret = fmax(ret, i - preval);
        } else {
            tmp = malloc(sizeof(UT_HASH));
            tmp->val = i;
            tmp->key = cnt;
            HASH_ADD_INT(hashTable, key, tmp);
        }
    }
 
    return ret;
}

想到其他语言的还有map，优先队列等接口，C语言真的太难了，遇到优先队列的题，基本死翘翘了。