【LeetCode: 525. 连续数组 + 前缀和 + 哈希表】

🌲 算法刷题专栏 | 面试必备算法 | 面试高频算法 🍀
🌲 越难的东西,越要努力坚持，因为它具有很高的价值，算法就是这样✨
🌲 作者简介：硕风和炜，CSDN-Java领域优质创作者🏆，保研|国家奖学金|高中学习JAVA|大学完善JAVA开发技术栈|面试刷题|面经八股文|经验分享|好用的网站工具分享💎💎💎
🌲 恭喜你发现一枚宝藏博主,赶快收入囊中吧🌻
🌲 人生如棋，我愿为卒，行动虽慢，可谁曾见我后退一步？🎯🎯

🚩 题目链接

• 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 <= 105
nums[i] 不是 0 就是 1

🌟 求解思路&实现代码&运行结果

⚡ 前缀和 + 哈希表

🥦 求解思路

1. 题目要求在一个二进制数组 nums 中，找到最长的连续子数组，使得子数组中 0 和 1 的数量相等。

2. 前缀和与哈希表：

   • 将数组中的 0 视为 -1，1 视为 1，计算前缀和 arr。

   • 使用哈希表 map 记录每个前缀和第一次出现的位置。

   • 如果某个前缀和已经存在于哈希表中，说明从 map.get(curSum) 到当前索引 i 的子数组中 0 和 1 的数量相等。

3. 更新最大长度：在遍历过程中，更新满足条件的子数组的最大长度。

4. 有了基本的思路，接下来我们就来通过代码来实现一下。

🥦 实现代码

class Solution {
    public int findMaxLength(int[] nums) {
        int n = nums.length;
        int[] prefixSum = new int[n + 1]; // 前缀和数组
        for (int i = 0; i < n; i++) {
            prefixSum[i + 1] = prefixSum[i] + (nums[i] == 0 ? -1 : 1); // 计算前缀和
        }
        HashMap<Integer, Integer> map = new HashMap<>(); // 哈希表记录前缀和的位置
        int max = 0; // 最大长度
        for (int i = 0; i <= n; i++) {
            int curSum = prefixSum[i];
            if (map.containsKey(curSum)) {
                max = Math.max(max, i - map.get(curSum)); // 更新最大长度
            } else {
                map.put(curSum, i); // 记录前缀和的位置
            }
        }
        return max; // 返回最大长度
    }
}

🥦 运行结果

💬 共勉