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;
}
相关推荐
এ᭄画画的北北2 小时前
力扣-139.单词拆分
算法·leetcode
完美的奶酪2 小时前
Leetcode-2537. 统计好子数组的数目
算法·leetcode
伊欧温2 小时前
最大公约数
c语言·算法
天机️灵韵3 小时前
谷歌时间序列算法:零样本预测如何重塑行业决策?
人工智能·python·算法·开源项目
k *3 小时前
数据结构--栈和队列
数据结构
飞火流云3 小时前
LSM树与B+树优缺点分析
数据结构·b树
代码的余温3 小时前
B树与B+树:数据库索引背后的秘密
数据结构·数据库·b树
zmuy4 小时前
148. 排序链表
数据结构·链表
W说编程4 小时前
算法导论第十四章 B树与B+树:海量数据的守护者
c语言·数据结构·b树·算法·性能优化
Jay Kay4 小时前
深入解析协程:高并发编程的轻量级解决方案
开发语言·c++·算法