Contiguous Array问题及解法

最新推荐文章于 2020-02-15 12:12:01 发布

我们要爱学习

最新推荐文章于 2020-02-15 12:12:01 发布

阅读量394

点赞数

分类专栏： LeetCode算法练习（Medium）文章标签： LeetCode Medium

本文链接：https://blog.csdn.net/u011809767/article/details/78132365

版权

LeetCode算法练习（Medium）专栏收录该内容

238 篇文章 1 订阅

订阅专栏

问题描述：

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

示例:

Input: [0,1]
Output: 2
Explanation: [0, 1] is the longest contiguous subarray with equal number of 0 and 1.

Input: [0,1,0]
Output: 2
Explanation: [0, 1] (or [1, 0]) is a longest contiguous subarray with equal number of 0 and 1.

问题分析：

这里我们重点看的是连续子数组0和1的数目zero和one，遍历时我们把zero - one 和 i 存储起来，若后续子数组中0的数目和1的数目相等，那么zero - one的值就会被检测到出现过一次，然后得到子数组的长度，最终比较得到更长的长度。

过程详见代码：

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