[LeetCode] 525. Contiguous Array

Posted aaronliu1991

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[LeetCode] 525. Contiguous Array相关的知识,希望对你有一定的参考价值。

连续数组。给定一个二进制数组, 找到含有相同数量的 0 和 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.

这个题目看似简单但是这个思路比较难想到。这个题的tag是hash table,感觉可以往前缀和为0的思路上去靠。题目问的是包含相同数量的0和1的子数组的长度,如果把0替换成-1,也就是说满足条件的子数组的和会是0。所以做法是创建一个hashmap记录前缀和他最后一次出现的位置的index。扫描的时候就按照加前缀和的方式做,当遇到某一个前缀和X已经存在于hashmap的时候,就知道这一段区间内的子数组的和为0了,否则这个前缀和不可能重复出现。举个例子,比如[0,1,0]好了,一开始存入了初始值0和他的下标-1,当遍历完0(-1)和1的时候,此时又得到这个prefix sum = 0,此时去看一下i - map.get(sum)的值是否能更大,以得出这个最长的子数组的长度。

时间O(n)

空间O(n)

Java实现

 1 class Solution {
 2     public int findMaxLength(int[] nums) {
 3         HashMap<Integer, Integer> map = new HashMap<>();
 4         map.put(0, -1);
 5         int res = 0;
 6         int count = 0;
 7 
 8         for (int i = 0; i < nums.length; i++) {
 9             count = count + (nums[i] == 1 ? 1 : -1);
10             if (map.containsKey(count)) {
11                 res = Math.max(res, i - map.get(count));
12             } else {
13                 map.put(count, i);
14             }
15         }
16         return res;
17     }
18 }

 

javascript实现

 1 /**
 2  * @param {number[]} nums
 3  * @return {number}
 4  */
 5 var findMaxLength = function(nums) {
 6     let map = new Map();
 7     map.set(0, -1);
 8     let res = 0;
 9     let sum = 0;
10     
11     for (let i = 0; i < nums.length; i++) {
12         sum += nums[i] == 0 ? -1 : 1;
13         if (map.has(sum)) {
14             res = Math.max(res, i - map.get(sum));
15         } else {
16             map.set(sum, i);
17         }
18     }
19     return res;
20 };

 

以上是关于[LeetCode] 525. Contiguous Array的主要内容,如果未能解决你的问题,请参考以下文章

[leetcode-525-Contiguous Array]

leetcode 525. Contiguous Array

leetcode 525. Contiguous Array

LeetCode 525: Continuous Array

leetcode-525连续数组

[LeetCode] 525. Contiguous Array