Day26-Contiguous Array(Medium)
问题描述:
Given a binary array, find the maximum length of a contiguous subarray with equal number of 0 and 1.
这道题在上个月的挑战里面就有了,求一个数组中最大子数组的长度,这个子数组中0,1的数量相同
Example:
Example 1:
Input: [0,1]
Output: 2
Explanation: [0, 1] is the longest contiguous subarray with equal number of 0 and 1.
Example 2:
Input: [0,1,0]
Output: 2
Explanation: [0, 1] (or [1, 0]) is a longest contiguous subarray with equal number of 0 and 1.
Note: The length of the given binary array will not exceed 50,000.
解法:
在代码里面标注了解释,因为上个月做过了,所以就不细解释了
class Solution:
def findMaxLength(self, nums: List[int]) -> int:
if len(nums) <= 1:
return 0
#对于求子数组的问题,需要时刻记着求累积和是一种很犀利的工具。这里需要用到一个 trick,遇到1就加1,遇到0,就减1,这样如果某个子数组和为0,就说明0和1的个数相等,这个想法真是太叼了
#累积和的解法:这道题要求解的是数组中最大的子数组的长度,这个子数组中包含0,1的个数相等。
#这时可以利用一个字典存储每个位置的累计和,遇到0时减1,遇到1时加1,这时如果当前位置的累积和为0,说明0和1的个数相等,同时如果出现新的数,将其存入hash表中,向后遍历遇到相同的情况时,减去hash表中的数据,就是中间的子数组的长度。同时这个子数组中0,1的个数也相同。
result = 0
sums = 0
temp = {}
temp[0] = -1
for i , n in enumerate(nums):
if n == 0:
sums -= 1
else:
sums += 1
if sums in temp:
result = max(result,i - temp[sums])
else:
temp[sums] = i
return result