【算法】前缀和(二)使用

文章目录

一、问题直化前缀和

1.拆拼

二、问题转用前缀和

1.模减消实质

2.同余定理

2.1证明

3.取模%

3.1计算机

[3.1.1向零取整 相除](#3.1.1向零取整 相除)

3.1.2符号

3.2数学

[3.2.1向下取整 相除](#3.2.1向下取整 相除)

3.2.2符号

3.3两者关系

3.3.1差异

3.3.2转化


上篇:【算法】前缀和(一)原理

一、问题直化前缀和

238. 除自身以外数组的乘积 - 力扣(LeetCode)

给你一个整数数组 nums,返回 数组 answer ,其中 answer[i] 等于 nums 中除 nums[i] 之外其余各元素的乘积 。

题目数据 保证 数组 nums之中任意元素的全部前缀元素和后缀的乘积都在 32 位 整数范围内。

不要使用除法, 且在 O(n) 时间复杂度内完成此题。

示例 1:

复制代码
输入: nums = [1,2,3,4]

输出: [24,12,8,6]

示例 2:

复制代码
输入: nums = [-1,1,0,-3,3]
输出: [0,0,9,0,0]

提示:

  • 2 <= nums.length <= 105
  • -30 <= nums[i] <= 30
  • 输入 保证 数组 answer[i]32 位 整数范围内

1.拆拼

整体 拆分能同类累积部分 拼合

java 复制代码
public int[] productExceptSelf(int[] nums) {
    int n = nums.length;
    int[] f = new int[n], g = new int[n], ret = new int[n];
    f[0] = g[n - 1] = 1;
    for (int i = 1; i < n; i++)
        f[i] = f[i - 1] * nums[i - 1];
    for (int i = n - 2; i >= 0; i--)
        g[i] = g[i + 1] * nums[i + 1];
    for (int i = 0; i < n; i++)
        ret[i] = f[i] * g[i];
    return ret;
}

二、问题转用前缀和

974. 和可被 K 整除的子数组 - 力扣(LeetCode)

给定一个整数数组 nums 和一个整数 k ,返回其中元素之和可被 k 整除的非空 子数组 的数目。

子数组 是数组中 连续 的部分。

示例 1:

复制代码
输入:nums = [4,5,0,-2,-3,1], k = 5
输出:7
解释:
有 7 个子数组满足其元素之和可被 k = 5 整除:
[4, 5, 0, -2, -3, 1], [5], [5, 0], [5, 0, -2, -3], [0], [0, -2, -3], [-2, -3]

示例 2:

复制代码
输入: nums = [5], k = 9
输出: 0

提示:

  • 1 <= nums.length <= 3 * 104
  • -104 <= nums[i] <= 104
  • 2 <= k <= 104

1.模减消实质

(a + p*k) % p = a % p


2.同余定理

差被++整++除 ++(a - b) % p = 0++ or (a - b) / p = k

=> 减数 模 除数 ++相等++ ++a % p = b % p++

2.1证明

(a - b) / p = k

-> a - b = p*k

-> a = b + p*k

-> a % p = (b + p*k) % p

-> a % p = b % p


3.取模%

3.1计算机

3.1.1向零取整 相除

余数绝对值 最接近0 地相除

3.1.2符号

同被除数


3.2数学

3.2.1向下取整 相除

余数正值 最接近0 地相除

3.2.2符号

正数


3.3两者关系

3.3.1差异

只有 负 % 正负数取模结果会不同 (计算机为负、数学为正)

3.3.2转化

计算机 相同等效、不同转化 为 数学 的模结果:(a % p + p) / p

java 复制代码
public int subarraysDivByK(int[] nums, int k) {
    Map<Integer,Integer> map = new HashMap<>();
    int sum = 0, ret = 0;
    map.put(0,1); // sum = k,sum % k = 0,自己整的区间可以被k整除的是结果,自己整的就是结果,但往前减区间来得 是得不到的,得特判为就是结果地 就开始累积
    for (int i = 0; i < nums.length; i++) {
        sum += nums[i];
        int r = (sum % k + k) % k;
        // 找满足的前缀和:
        ret += map.getOrDefault(r,0); // (sum - x) % k = 0 => sum % k = x % k,往前找 已存进哈希表里的 前缀和x % k的模 有多少个 等于 当前总区间前缀和sum % k的模
        // 存前缀和:
        map.put(r, map.getOrDefault(r,0) + 1);
    }
    return ret;
}
相关推荐
ChaHae-In4 分钟前
MyBatis入门操作
java·mybatis
闲研随记6 分钟前
【文献阅读 ICLR 2026】RL算法:Co-Rewarding
算法
xcLeigh9 分钟前
Go入门:短变量声明的陷阱与最佳实践
java·redis·golang·教程·变量
青山木15 分钟前
Hot 100 --- 最小栈
java·数据结构·算法·leetcode
疯狂打码的少年18 分钟前
【数据结构】栈的应用:表达式求值(后缀表达式)
java·数据结构·笔记·算法
szephyr24 分钟前
腾讯云 ADP 智能体的 Skills 版本回滚总是回到旧配置,是缓存没清还是版本管理没开?
java·缓存·腾讯云
云烟成雨TD28 分钟前
Micrometer 系列【39】链路追踪:入门案例 | 环境准备
java·云原生·链路追踪
HiDev_29 分钟前
【非标自动化】plc执行顺序问题、扫描周期问题
深度学习·算法·机器学习
chuan.bai29 分钟前
Java RAG 实战(第 3 篇):从交互式聊天到多轮上下文
java·人工智能·macos·ai
9523643 分钟前
Sentinel
java·后端·spring·sentinel·springcloud