leetcode 525. 连续数组

这段代码实现了用前缀和和哈希表寻找0和1数量相同的最长子数组。核心思路是将0视为-1,然后计算前缀和,利用哈希表记录每个前缀和第一次出现的位置,当遇到相同前缀和时计算长度并更新最大值。

关键步骤:

  1. 初始化哈希表,记录前缀和0的初始位置为-1
  2. 遍历数组,遇到1时前缀和加1,遇到0时前缀和减1
  3. 检查当前前缀和是否已存在哈希表中:
    • 若存在,计算当前位置与首次出现位置的距离,更新最大长度
    • 若不存在,记录当前前缀和的首次出现位置
  4. 返回最大长度

示例输入[0,1,0,1]的输出为4,对应整个数组。时间复杂度O(n),空间复杂度O(n)。

cpp 复制代码
#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;

int findMaxLength(vector<int>& nums) {
    int maxLen = 0;
    int counter = 0;
    // 哈希表记录前缀和第一次出现的位置
    unordered_map<int, int> hashmap;
    // 初始化前缀和为0的位置为-1
    hashmap[0] = -1;

    for (int i = 0; i < nums.size(); i++) {
        // 遇到1加1,遇到0减1
        if (nums[i] == 1) {
            counter++;
        } else {
            counter--;
        }

        // 如果当前前缀和已经存在,计算当前位置到首次出现位置的长度
        if (hashmap.find(counter) != hashmap.end()) {
            int len = i - hashmap[counter];
            if (len > maxLen) {
                maxLen = len;
            }
        } else {
            // 记录前缀和首次出现的位置
            hashmap[counter] = i;
        }
    }

    return maxLen;
}

int main() {
    vector<int> nums = {0, 1, 0, 1};
    cout << "最长子数组长度: " << findMaxLength(nums) << endl;
    return 0;
}
相关推荐
敲代码的瓦龙29 分钟前
C++?继承!!!
c语言·开发语言·c++·windows·后端·算法
简简单单做算法32 分钟前
基于FPGA的二叉决策树cart算法verilog实现,训练环节采用MATLAB仿真
算法·决策树·fpga开发·cart算法·二叉决策树
白熊18844 分钟前
【机器学习基础】机器学习入门核心算法:K-近邻算法(K-Nearest Neighbors, KNN)
算法·机器学习·近邻算法
北京地铁1号线1 小时前
MMdetection推理验证输出详解(单张图片demo)
前端·算法
oioihoii1 小时前
C++23 新成员函数与字符串类型的改动
算法·c++23
似水এ᭄往昔1 小时前
【数据结构】——二叉树堆(下)
数据结构·算法
GG不是gg2 小时前
Prim算法剖析与py/cpp/java语言实现
算法
秋山落叶万岭花开ღ2 小时前
探索数据结构之顺序表:从入门到精通
数据结构·python·算法
这张生成的图像能检测吗2 小时前
OpenGAN:基于开放数据生成的开放集识别
人工智能·pytorch·深度学习·算法·机器学习·生成对抗网络·聚类
夕泠爱吃糖3 小时前
红黑树,B树,二叉树之间的不同
数据结构·算法·不同树的区别