525. Contiguous Array 题解

525. Contiguous Array 题解

题目描述：

Given a binary array, find the maximum length of a contiguous subarray with equal number of 0 and 1.

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.

题目链接：525. Contiguous Array

算法描述：

          由题意知，给定一串由 0 和 1 组成的数组，我们需要找到这串数组中最长的连续子数组，且这个连续子数组中 0 和 1 出现的次数一样。

       我们按照这样的思路来做：首先遍历这个原数组，将原数组中所有的 0 转换为 -1，这样我们就可以认为在元素相加时，如果结果为 0 ，则这段元素中的 0 和 1 个数相同。之后，我们需要构造一个 hashmap：mp。mp 的 key 存放的是数组的元素和（从头元素加到此位置元素）， value 存放的是位置索引。并且，mp[0] = -1。

        之后，我们对这个数组进行遍历，从头元素依次相加，当发现正在遍历元素 i 的 mp 的 key 与之前便利的元素（设为 j ）一样时，说明从 i 到 j 这段的元素相加之和为 0。我们取最大的这段距离。

        按照上面的方法，我们遍历完所有元素即可。

代码：

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int ans = 0;
        int sum = 0;
        for(int i=0; i<nums.size(); i++){
            if(nums[i] == 0){
                nums[i] = -1;
            }
        }
        
        map<int,int>mp;
        mp[0] = -1;
        for(int i=0; i<nums.size(); i++){
            sum += nums[i];
            if(mp.find(sum) != mp.end()){
                ans = max(ans, i-mp[sum]);
            }
            else{
                mp[sum] = i;
            }
        }
        return ans;
    }
};