525. 连续数组

完美代码

于 2021-06-03 15:35:01 发布

阅读量293

点赞数 1

CC 4.0 BY-SA版权

谢绝转载-https://update.blog.youkuaiyun.com

本文链接：https://blog.youkuaiyun.com/update7/article/details/117524641

编程. 同时被 2 个专栏收录

578 篇文章 ¥299.90 ¥399.90

订阅专栏

编程..

570 篇文章 ¥299.90 ¥399.90

订阅专栏

给定一个二进制数组 nums，目标是找到0和1数量相等的最长连续子数组。例如，对于输入 [0,1]，输出为2，因为 [0, 1] 是符合条件的最长子数组。" 6724835,753823,寻找单链表环的入口点：算法解析与实现,"['算法', '编程', 'C']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个二进制数组 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

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int pre=0,p;
        map<int,int> mp;
        mp[0]=-1;
        int ret = 0;
        for(int i = 0;i<nums.size();++i)
        {
            pre = p = pre+(nums[i]==1?1:-1);
            auto it = mp.find(p);
            if(it!=mp.end())
            {
                ret = max(ret,i-(it->second));
            }
            else
            mp[p] = i;
        }
        return ret;
    }
};