字节面试原题，两人都没做出来，都凉了

将 脚本之家 设为“星标⭐”第一时间收到文章更新本文经数据结构和算法（id:sjjghsf）授权转载

这题是LeetCode的第525题：连续数组。有两位网友在字节的面试中都遇到这题，不过很遗憾的是他俩都没做出来，至于后来有没有被录取就不知道了。这道题是一道中等难度的题，按说不算太难，下面我们就来看下。

问题描述

来源：LeetCode第525题
难度：中等

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

示例1：

输入: nums = [0,1]

输出: 2

说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。

示例2：

输入: nums = [0,1,0]

输出: 2

说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

• 1 <= nums.length <= 10^5

• nums[i] 不是 0 就是 1

问题分析

这题实际上是让求一个最长的子数组，并且这个子数组中0和1的数量必须相同。因为数组中的元素只能是0或1，直接计算不太好解，我们可以换种思路，把数组中的0换成-1。

这样0和1的数量必须相同就变成了-1和1的数量必须相同了，他们相加肯定是0，所以这题就变成了求一个最长的子数组，并且这个子数组的和必须是0。

我们以示例2为例来看下，改变之后的数组就变成了[-1,1,-1]。那么这题的解题思路就很明确了，我们可以直接使用前缀和，对改变之后的数组进行累加，然后使用一个map对累加的值进行存储，因为是求最大长度，如果遇到相同的值则不能覆盖。

累加的时候如果出现相同的值，只需要用当前值的下标减去前面相同值的下标，中间这段就是和为0的子数组，我们只需要保存他的长度，记录最大值即可。

这里还要注意使用map存储前缀和的时候，要考虑0的情况，也就是说如果从数组的第一个元素到当前元素nums[i]的和是0，那么这个长度就是i+1，因为数组的下标是从0开始的，所以这里前缀和为0的时候，我们给它一个默认值-1。

JAVA：

public int findMaxLength(int[] nums) {
    Map<Integer, Integer> map = new HashMap<>();
    map.put(0, -1);// 前缀和为0的时候，给一个默认值-1。
    int preSum = 0, max = 0;
    for (int i = 0; i < nums.length; i++) {
        // 计算前缀和，原数组中如果是0，就让它变成-1。
        preSum += nums[i] == 0 ? -1 : 1;
        if (map.containsKey(preSum)) {
            // 如果出现相同的前缀和，就计算长度，并保存最大值。
            max = Math.max(max, i - map.get(preSum));
        } else {
            // 如果没有出现相同的前缀和，就把它存起来。
            map.put(preSum, i);
        }
    }
    return max;
}

C++:

public:
    int findMaxLength(vector<int>& nums) {
        unordered_map<int, int> mp ;
        mp[0]= -1;// 前缀和为0的时候，给一个默认值-1。
        int preSum = 0, maxLength = 0;
        for (int i = 0; i < nums.size(); i++) {
            // 计算前缀和，原数组中如果是0，就让它变成-1。
            preSum += nums[i] == 0 ? -1 : 1;
            if (mp.count(preSum)) {
                // 如果出现相同的前缀和，就计算长度，并保存最大值。
                maxLength = max(maxLength, i - mp[preSum]);
            } else {
                // 如果没有出现相同的前缀和，就把它存起来。
                mp[preSum]= i;
            }
        }
        return maxLength;
    }

  推荐阅读：