LeetCode算法心得——和可被 K 整除的子数组(前缀和+HashMap)

2023-09-14 18:53:38

大家好,我是晴天学长,同余定理的应用,需要的小伙伴可以关注支持一下哦!后续会继续更新的。


1) .和可被 K 整除的子数组

在这里插入图片描述
题目描述

给定一个整数数组 A,返回其中元素之和可被 K 整除的(连续、非空)子数组的数目。

示例:

输入:A = [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


2) .算法思路

前缀和+HashMap

解析:

我们需要求出满足条件的区间,见下图
在这里插入图片描述
我们需要找到满足,和为 K 的区间。我们此时 presum 是已知的,k 也是已知的,我们只需要找到 presum - k区间的个数,就能得到 k 的区间个数。但是我们在当前题目中应该怎么做呢?见下图。

在这里插入图片描述
我们在之前的例子中说到,presum[j+1] - presum[i] 可以得到 nums[i] + nums[i+1]+… nums[j],也就是[i,j]区间的和。

那么我们想要判断区间 [i,j] 的和是否能整除 K,也就是上图中紫色那一段是否能整除 K,那么我们只需判断

(presum[j+1] - presum[i] ) % k 是否等于 0 即可,

我们假设 (presum[j+1] - presum[i] ) % k == 0;则

presum[j+1] % k - presum[i] % k == 0;

presum[j +1] % k = presum[i] % k ;

我们 presum[j +1] % k 的值 key 是已知的,则是当前的 presum 和 k 的关系,我们只需要知道之前的前缀区间里含有相同余数 (key)的个数。则能够知道当前能够整除 K 的区间个数。见下图
在这里插入图片描述
我们看到上面代码中有一段代码是这样的
int key = (presum % K + K) % K;
这是为什么呢?不能直接用 presum % k 吗?

这是因为当我们 presum 为负数时,需要对其纠正。纠正前(-1) %2 = (-1),纠正之后 ( (-1) % 2 + 2) % 2=1 保存在哈希表中的则为 1.则不会漏掉部分情况,例如输入为 [-1,2,9],K = 2如果不对其纠正则会漏掉区间 [2] 此时 2 % 2 = 0,符合条件,但是不会被计数。

那么这个题目我们可不可以用数组,代替 map 呢?当然也是可以的,因为此时我们的哈希表存的是余数,余数最大也只不过是 K-1所以我们可以用固定长度 K 的数组来模拟哈希表。


3) .算法步骤

1.创建一个HashMap对象 map 用于存储余数与对应出现次数的映射关系。
2.初始化变量 PerSum 为0,表示当前位置的前缀和。
3.初始化变量 answer 为0,表示满足条件的子数组个数。
4.将余数为0的初始情况放入 map 中,即前缀和为0的情况,出现次数为1。
5.使用一个循环遍历数组中的元素。
6.将当前元素的值累加到 PerSum 中,即计算当前位置的前缀和。
7.使用同余定理,计算当前前缀和除以k的余数,由于负数取模的结果可能为负数,为8.确保结果为非负整数,需要进行加k再取模的操作。
9.检查 map 中是否包含当前余数,如果是,则将对应的出现次数加到 answer 中。
10.将当前余数和对应的出现次数放入 map 中,更新映射关系。
11.循环结束后,返回 answer,表示满足条件的子数组个数。


4).代码示例

class Solution {
    public int subarraysDivByK(int[] nums, int k) {
         Map<Integer, Integer> map = new HashMap<Integer, Integer>();
            long PerSum = 0;
            int answer = 0;
            map.put(0, 1);
            for (int i = 0; i < nums.length; i++) {
                PerSum += nums[i];
                // 同余定理,因为不符合分配律
                int mod = (int) ((PerSum%k+k)%k);
                if (map.containsKey(mod)) {
                    answer+=map.get(mod);
                }
                map.put(mod, map.getOrDefault(mod, 0) + 1);
            }
            return answer;
    }
}

5).总结

  • 同余定理的运用。

试题链接:

更多推荐

【树形 DP】如何从“方向“角度理解树形 DP

题目描述这是LeetCode上的「834.树中距离之和」,难度为「困难」。Tag:「树形DP」、「DFS」、「动态规划」、「树」给定一个无向、连通的树。树中有n个标记为0...n-1的节点以及n-1条边。给定整数n和数组edges,表示树中的节点和之间有一条边。返回长度为n的数组answer,其中answer[i]是树

UML六大关系总结

UML六大关系有:继承、关系、聚合、组合、实现、依赖。分为通过图和代码总结这些关系。1、继承classBird:Animal{}说明:一段都是子类继承父类,在子类的后面用一个冒号表示,冒号后面跟着父类的名字。继承只能继承父类共有和保护的属性或方法,私有的变量或方法不能被子类继承。2、关联ClassPenguin{pri

基于Pandas+余弦相似度+大数据智能护肤品推荐系统——机器学习算法应用(含Python工程源码)+数据集

目录前言总体设计系统整体结构图系统流程图运行环境Python环境Pycharm环境模块实现1.文件读入2.推荐算法1)数据预处理2)计算相似度3)排序并提取产品4)组合推荐算法3.应用模块1)得到最终产品2)筛选过敏物质3)筛选相互禁忌的产品4)输出单品推荐与组合推荐4.测试调用函数系统测试工程源代码下载其它资料下载前

Leetcode.2591 将钱分给最多的儿童

题目链接Leetcode.2591将钱分给最多的儿童rating:1531题目描述给你一个整数moneymoneymoney,表示你总共有的钱数(单位为美元)和另一个整数childrenchildrenchildren,表示你要将钱分配给多少个儿童。你需要按照如下规则分配:所有的钱都必须被分配。每个儿童至少获得111美

第九天:QT入门保姆教程(常用的控件,信号与槽,定时器 QTimer,样式表 Qt Style Sheets,sqlite3数据库,开发板串口)

QT的简介我另外分享了一个qt案例源码包,里面包括文章中的任务源码和一系列常用案例需要的点击此处下载官网www.qt.ioQT是一个基于C++的跨平台的应用程序开发框架跨平台:一次编写,到处编译主流的平台都支持,如:Windows,Linux,Android,MacOS...应用程序:主要用于GUI程序开发,也可以用于

RT-Thread(学习)

RT-Thread是一款完全由国内团队开发维护的嵌入式实时操作系统(RTOS),具有完全的自主知识产权。经过16个年头的沉淀,伴随着物联网的兴起,它正演变成一个功能强大、组件丰富的物联网操作系统。RT-Thread概述RT-Thread,全称是RealTime-Thread,顾名思义,它是一个嵌入式实时多线程操作系统,

2023年最全详解:什么是销售管理?销售管理必备百科指南!

销售管理是什么?怎么样进行销售管理?都有哪些销售管理类型?销售管理都有哪些职责?如何高效的进行销售管理?本篇,我们将带大家深入浅出的了解销售管理,并且通过crm客户管理系统将销售管理的作用发挥到最大化!一、什么是销售管理?销售管理是现代企业不可或缺的重要部分。它涵盖了企业销售过程中的所有活动,包括市场营销、销售策略、销

企业商标信息API:品牌管理的秘密武器

引言当今数字时代,品牌管理变得比以往任何时候都更具挑战性。企业需要不断创新、保护知识产权、实时监测市场动态以及应对竞争压力。在这个竞争激烈的环境中,企业商标信息API已经成为品牌管理的秘密武器,为企业提供了无可估量的价值。企业商标信息API的作用企业商标信息API是一种应用程序接口,它允许企业访问商标数据库中的关键数据

生产制造业厂家固定资产怎么管理

固定资产的管理对于企业的运营效率和盈利能力具有重要影响。然而,传统的固定资产管理方法往往存在许多问题,如资产的低效使用、维护成本高昂以及决策者对资产价值缺乏准确了解等。&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;因此,我们需要采用一种全新的方式来管理我们的固定资产。本

MySQL常见面试题(一)

😀前言在数据库管理系统中,存储引擎起着核心的角色,它决定了数据管理和存储的方式。MySQL作为一个领先的开源关系型数据库管理系统,提供了多种存储引擎来满足不同的需求和优化不同的应用。除了选择合适的存储引擎,数据库的设计还涉及到范式设计和表设计,这两者都对数据库的性能和数据一致性有深远的影响。在本文中,我们将探讨MyS

清易低功耗智能雨量监测站概述

一、低功耗智能雨量监测站概述产品概述低功耗智能雨量监测站基于智能传感、无线通信、智能处理与智能控制等物联网技术的开发,利用智能传感技术,通过传感器测量降雨量,并使用物联网进行传输。无需专门的通信线路,在联网的状态下,数据可快速、主动的上报到云平台,用户可在电脑或手机,随时随地浏览数据。二、技术参数测量参数降雨量◇测量范

热文推荐