Leetcode--Java--525. 连续数组

题目描述

给定一个二进制数组 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. 将所有的0换成-1，可以将问题转换为求最长的和为0的子数组
2. 快速求一个区间的和，可以用前缀和数组。在求一段子数组的和可以表示为两个前缀和之差是否等于0，然后用哈希表的话，相当于看是否存在其中一个，只要其中一个存在，就存在整个。思路类似两数之和中对于每个x先找 target - x是否存在。这里是对于每个前缀和数组f[i] 是否存在f[j - 1]，使得f[i] - f[j - 1] == 0，用哈希表也就是判断f[j - 1]是否存在