[leetcode 前缀和]

简介: [leetcode 前缀和]

525. 连续数组 M

:::details

给定一个二进制数组 nums , 找到含有相同数量的 01 的最长连续子数组,并返回该子数组的长度。

示例 1:

输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。

示例 2:

输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

提示:

  • 1 <= nums.length <= 105
  • nums[i] 不是 0 就是 1

解题思路

因为只会出现0或1,求相同数量的最长连续子数组,所以为了方便,我们把0定义为-1,当前缀和等于0时,说明,当前子数组的01相等。

func findMaxLength(nums []int) (maxLength int) {
  n := len(nums)
  /**
  记录前缀和出现的下标
  */
  hash := map[int]int{0: -1}
  k := 0
  for i := 0; i < n; i++ {
    if nums[i] == 0 {
      k--
    } else {
      k++
    }
    if prevIndex, ok := hash[k]; ok {
      maxLength = max(maxLength, i-prevIndex)
    } else {
      hash[k] = i
    }
  }
  return maxLength
}
func max(a, b int) int {
  if a > b {
    return a
  }
  return b
}

:::

523. 连续的子数组和 - 力扣(LeetCode)M

:::details

给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:

子数组大小 至少为 2 ,且

子数组元素总和为 k 的倍数。

如果存在,返回 true ;否则,返回 false 。

如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。0 始终视为 k 的一个倍数。

示例 1:

输入:nums = [23,2,4,6,7], k = 6

输出:true

解释:[2,4] 是一个大小为 2 的子数组,并且和为 6 。

示例 2:

输入:nums = [23,2,6,4,7], k = 6

输出:true

解释:[23, 2, 6, 4, 7] 是大小为 5 的子数组,并且和为 42 。

42 是 6 的倍数,因为 42 = 7 * 6 且 7 是一个整数。

示例 3:

输入:nums = [23,2,6,4,7], k = 13

输出:false

提示:

1 <= nums.length <= 105

0 <= nums[i] <= 109

0 <= sum(nums[i]) <= 231 - 1

1 <= k <= 231 - 1

来源:力扣(LeetCode)

链接:https://leetcode.cn/problems/continuous-subarray-sum

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

因为题目要求的是子数组元素总和是k的倍数,也就是说,需要取模运算。

所以,在求前缀和的时候,直接求余数,当出现相同余数的时候,说明当前子数组的前缀和符合倍数要求,然后判断子数组长度,如果符合条件则直接返回。

func checkSubarraySum(nums []int, k int) bool {
  n := len(nums)
  if n < 2 {
    return false
  }
  /**
  规定空的前缀的结束下标为 -1,
  由于空的前缀的元素和为 0,因此在哈希表中存入键值对 (0,-1)。
  */
  prevSum := map[int]int{0: -1}
  remainder := 0
  for i, num := range nums {
    remainder = (remainder + num) % k
    if prevIndex, ok := prevSum[remainder]; ok {
      if i-prevIndex >= 2 {
        return true
      }
    } else {
      prevSum[remainder] = i
    }
  }
  return false
}

:::

相关文章
|
算法 Android开发 索引
LeetCode 周赛上分之旅 #44 同余前缀和问题与经典倍增 LCA 算法
学习数据结构与算法的关键在于掌握问题背后的算法思维框架,你的思考越抽象,它能覆盖的问题域就越广,理解难度也更复杂。在这个专栏里,小彭与你分享每场 LeetCode 周赛的解题报告,一起体会上分之旅。
81 0
|
5月前
【LeetCode刷题】前缀和解决问题:742.寻找数组的中心下标、238.除自身以外数组的乘积
【LeetCode刷题】前缀和解决问题:742.寻找数组的中心下标、238.除自身以外数组的乘积
LeetCode-798 得分最高的最小论调 及差分和前缀和的学习
LeetCode-798 得分最高的最小论调 及差分和前缀和的学习
|
6月前
|
自然语言处理 索引
leetcode-745:前缀和后缀搜索
leetcode-745:前缀和后缀搜索
60 0
|
算法 索引
LeetCode算法小抄--数组(双指针、差分数组、前缀和)
LeetCode算法小抄--数组(双指针、差分数组、前缀和)
|
人工智能 算法 搜索推荐
LeetCode 周赛 338,贪心 / 埃氏筛 / 欧氏线性筛 / 前缀和 / 二分查找 / 拓扑排序
大家好,我是小彭。 上周末是 LeetCode 第 338 场周赛,你参加了吗?这场周赛覆盖的知识点很多,第四题称得上是近期几场周赛的天花板。
111 0
|
算法 索引
leetcode-每日一题745. 前缀和后缀搜索(哈希和字典树)
如果我们用前缀 prefix 和 后缀 suff去暴力对比所有单词肯定会超时,我们可以先把单词里所有的前缀后缀组合,中间用特殊符号@连接,对应的最大下标存入哈希表中。搜索时,用特殊符号@连接前缀后缀,在哈希表中进行搜索
96 0
leetcode-每日一题745. 前缀和后缀搜索(哈希和字典树)
|
算法
LeetCode 周赛 340,质数 / 前缀和 / 极大化最小值 / 最短路 / 平衡二叉树
今天讲 LeetCode 单周赛第 340 场,今天状态不好,掉了一波大分。
104 0
|
存储 C++
【力扣·每日一题】689. 三个无重叠子数组的最大和 (C++ 前缀和优化dp 保存路径)
【力扣·每日一题】689. 三个无重叠子数组的最大和 (C++ 前缀和优化dp 保存路径)
104 0
【力扣·每日一题】689. 三个无重叠子数组的最大和 (C++ 前缀和优化dp 保存路径)
力扣每日一题:523.连续的子数组和 前缀和+哈希表解法
力扣每日一题:523.连续的子数组和 前缀和+哈希表解法
133 0