525. Contiguous Array 题解

xiaoxiao2021-02-28  59

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; } };

转载请注明原文地址: https://www.6miu.com/read-76907.html

最新回复(0)