Leetcode算法学习日志-525 Contiguous Array

Leetcode 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.

题意分析

给一串由0和1构成的数组，求其中0,1个数相同最小数组的长度。

解法分析

当遍历到0时都看做-1，对所有的元素进行累加，如果加和为0，则说明前面整个子数组都是0,1相等的子数组，如果和不为0，则将该和作为key同相应下标放入map中，若发现加和key存在，则表明从第一次加和为key时到第二次加和为key时的元素加和为0，间隔子数组为0,1平衡数组，用两次的下标相减就能得到相应长度，与maxLen比较，取大值。C++代码如下：

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

改程序的算法复杂度为O(n)。