【LeetCode 30天挑战活动】Day 13. Contiguous Array

题目描述

给定一个二进制数组, 找到含有相同数量的 0 和 1 的最长连续子数组（的长度）。
示例：
输入: [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

解题思路

暴力法是第一反应且非常直观，但毕竟不到万不得已最好是不要用暴力法，于是开始思考有没有更好的解决方案。
这种找最长子数组的题很多可以使用滑动窗口，但这题比较特殊，因为窗口右侧不到最后一个元素是没有办法确定是否找到 / 没找到符合题意的子数组，所以滑动窗口其实和暴力法相差不大。
想了很久确实没有比较好的思路，于是就去看了题解。题解用了一个非常巧妙的方法，记录如下：
维护一个变量count，遍历数组，当遇到0时，count--，当遇到1时，count++。
那么如何判断0和1数量相同呢？当count出现重复。
以[0 0 1 0 0 0 1 1]为例，count的变化如下：

在A、B、C三点处，count的值均为-2，也就是说从A到B，从B到C，从A到C，这三段中，0和1的数量是相等的，那么自然，最长的子数组就是从A到C。
根据上述分析，由于我们只要找最长子数组，那么就只需要一个额外的数据结构来存放count和它第一次出现时的对应位置，之后再出现相同的count时只要和第一次出现的位置一起计算出长度。显然使用hashMap是最适合的。
代码如下：

class Solution {
    public int findMaxLength(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        
        int count = 0;
        int maxLen = 0;
        Map<Integer, Integer> map = new HashMap<>();
        
        // 注意要将初始count为0时的情况放入map中
        map.put(0, -1);
        
        for (int i = 0; i < nums.length; ++i) {
            if (nums[i] == 0) {
                count--;
            }
            else {
                count++;
            }

			// 上面这个if-else可以简单写成：count = count + (nums[i] == 1 ? 1 : -1);
            
            if (map.containsKey(count)) {
                maxLen = Math.max(maxLen, i - map.get(count));
            }
            else {
                map.put(count, i);
            }
        }
        
        return maxLen;
        
    }
}

反思总结

这道题目最需要学习的思想就是，“当count值相等时，意味着中间遇到了相等数量的0和1”
我之前看到0和1的时候就试图往位运算上想，但没有想出比较好的方案，不过这个count何尝不是一种和位运算异曲同工的思想呢。数组中只有0和1两个对立的元素，那么它们就给count带来完全相反的两种效果，当效果抵消，说明数量相等，这是一个很自然的转化。
我还是见的题太少了（叹