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: 2Explanation: [0, 1] is the longest contiguous subarray with equal number of 0 and 1.

Example 2:

Input: [0,1,0]Output: 2Explanation: [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)。