4月30天leetcode训练-Day13

Day13-Contiguous Array

问题描述：

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.

解法：

有一个博主说过：遇到子数组的问题，要时刻牢记累积和是一种很犀利的工具。虽然我还不怎么会用这个工具，但是这道题也是可以用累积和来进行求解的，遍历数组，初始sum = 0,遇到0则sum -= 1,遇到1则sum+= 1,然后把每个位置的sum存入字典中。如果sum = 0说明当前位置之前的子数组中0和1的个数相等，则更新结果result，这里也可以解释为什么初始化字典{0，-1}的映射，这里就表示从初始位置开始的子数组满足条件的情况，而如果再遍历的过程中遇到sum出现过的情况时则用当前位置减去字典中的值得到中间数组的长度，这个中间数组中0，1的个数也是相同的。这里要记住遇到相同sum时不要更新字典中的值，因为要找最长的子数组所以要保留第一次出现的位置就行了。

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

时间复杂度为O（N），空间复杂度为O（N）