想要精通算法和SQL的成长之路 - 连续的子数组和

想要精通算法和SQL的成长之路 - 连续的子数组和

  • 前言
  • [一. 连续的子数组和](#一. 连续的子数组和)
    • [1.1 最原始的前缀和](#1.1 最原始的前缀和)
    • [1.2 前缀和 + 哈希表](#1.2 前缀和 + 哈希表)

前言

想要精通算法和SQL的成长之路 - 系列导航

一. 连续的子数组和

原题链接

1.1 最原始的前缀和

如果这道题目,用前缀和来算,我们的思路一般是这样:

  1. 计算这个数组的前缀和。
  2. 循环遍历数组的每个元素,以每个元素作为起点,向后寻找第二个元素(索引至少是起点+2)作为终点,计算两者的区域和。再判断是否满足条件。

那么这样的代码写出来就是这样:

java 复制代码
public boolean checkSubarraySum(int[] nums, int k) {
    int n = nums.length;
    // 计算前缀和
    int[] preSum = new int[n + 1];
    for (int i = 0; i < n; i++) {
        preSum[i + 1] = preSum[i] + nums[i];
    }
    // i 遍历到 preSum。length -2 即可,
    for (int i = 0; i < n - 1; i++) {
        // 区间长度 > 2
        for (int j = i + 2; j <= n; j++) {
            // 前缀和差即是[i,j]之间的区域和
            int diff = preSum[j] - preSum[i];
            if (diff % k == 0) {
                return true;
            }

        }
    }
    return false;
}

结果如下:

1.2 前缀和 + 哈希表

我们从这段代码入手:

java 复制代码
int diff = preSum[j] - preSum[i];
if (diff % k == 0) {
    return true;
}

即:

  • (preSum[j] - preSum[i] ) % k = 0;
  • preSum[j] % k == preSum[i] % k;

那么我们只需要利用哈希表,记录每个前缀和对于k的一个取模值是多少即可,1. 存储的是它们的下标。

  1. 如果遇到取模值相同的,并且两个下标差 > 2,就满足条件。

那么代码优化:

java 复制代码
public boolean checkSubarraySum(int[] nums, int k) {
    int n = nums.length;
    // 计算前缀和
    int[] preSum = new int[n + 1];
    for (int i = 0; i < n; i++) {
        preSum[i + 1] = preSum[i] + nums[i];
    }
    HashSet<Integer> set = new HashSet<>();
    for (int i = 2; i <= n; i++) {
        set.add(preSum[i - 2] % k);
        if (set.contains(preSum[i] % k)) {
            return true;
        }
    }
    return false;
}
相关推荐
爱喝矿泉水的猛男2 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
YuTaoShao2 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
源码_V_saaskw3 小时前
JAVA图文短视频交友+自营商城系统源码支持小程序+Android+IOS+H5
java·微信小程序·小程序·uni-app·音视频·交友
超浪的晨3 小时前
Java UDP 通信详解:从基础到实战,彻底掌握无连接网络编程
java·开发语言·后端·学习·个人开发
YouQian7723 小时前
Traffic Lights set的使用
算法
双力臂4043 小时前
Spring Boot 单元测试进阶:JUnit5 + Mock测试与切片测试实战及覆盖率报告生成
java·spring boot·后端·单元测试
Edingbrugh.南空4 小时前
Aerospike与Redis深度对比:从架构到性能的全方位解析
java·开发语言·spring
go54631584654 小时前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法
QQ_4376643144 小时前
C++11 右值引用 Lambda 表达式
java·开发语言·c++
永卿0014 小时前
设计模式-迭代器模式
java·设计模式·迭代器模式