面试题11：0和1个数相同的子数组（Java版）

题目：输入一个只包含0和1的数组，请问如何求0和1的个数相
同的最长连续子数组的长度？例如，在数组[0，1，0]中有两个子数
组包含相同个数的0和1，分别是[0，1]和[1，0]，它们的长度都是
2，因此输出2。

// 用-1来代替数组中的0，因为这样的话当0和1的数量相等时，
// 连续子数组中的和就恰好为零，因此当m>n，且前m个数和前n个数的和都相同时，
// 前m个数减去前n个数则为0，即第n+1个数到第m个数之间的数的和为零0
// 即第n+1个数到第m个数这个连续子数组就为0和1数量相同的连续子数组
// 在哈希表中用key来表示连续的数求和的值，value来表示这段连续的数的最后一个数的索引
// 即当key相同时value的差值就是连续子数组的长度
public static int findMaxLength(int[] nums) {
    Map<Integer, Integer> sumToIndex = new HashMap<>();
    // 当sum为零时，即代表从头开始的n个数中0和1的个数相等
    // 因为从头开始的连续子数组长度 = 尾索引 + 1
    // （对应到下面的 i - sumToIndex.get(sum) 中，则是长度 = i - (-1))
    // 哈希表中key为0的数value初始为-1
    sumToIndex.put(0, -1);
    int sum = 0;
    int maxLength = 0;
    for (int i = 0; i < nums.length; i++) {
        sum += nums[i] == 0 ? -1 : 1;
        // 如果哈希表中存在与该条连续子数组的和相等的连续子数组
        // 则将两个连续子数组之间的索引相减便得出
        // 0和1个数相等的连续子数组长度
        // 否则在哈希表中添加这条key为该连续子数组的和，
        // value为该连续子数组最后一个数索引的记录
        if (sumToIndex.containsKey(sum)) {
            maxLength = Math.max(maxLength, i - sumToIndex.get(sum));
        } else {
            sumToIndex.put(sum, i);
        }
    }
    return maxLength;
}