leetcode------523. 连续的子数组和[1]
本文共 1653 字，大约阅读时间需要 5 分钟。
为了解决这个问题，我们需要判断给定数组中是否存在至少两个连续的子数组，其和为目标整数 k 的倍数。我们可以通过前缀和和模运算来高效地解决这个问题。
方法思路
特殊情况处理：当目标整数 k 为 0 时，我们需要检查数组中是否存在连续的两个 0，因为只有这样，子数组的和才能为 0。 前缀和和模运算：对于非零的 k，我们可以使用前缀和数组来记录每个位置的和，并使用模运算来记录每个前缀和对 k 取模的结果。这样，当我们发现两个前缀和对 k 取模后的结果相同时，说明它们之间的子数组和是 k 的倍数。 解决代码
#include #include using namespace std;class Solution {public: bool checkSubarraySum(vector & nums, int k) { int n = nums.size(); if (n < 2) return false; // 子数组至少需要2个元素 if (k == 0) { // 检查是否存在连续的两个0 for (int i = 0; i < n - 1; ++i) { if (nums[i] == 0 && nums[i + 1] == 0) { return true; } } return false; } else { vector prefix(n + 1, 0); for (int i = 1; i <= n; ++i) { prefix[i] = prefix[i - 1] + nums[i - 1]; } unordered_map seen; seen[prefix[0] % k] = 0; for (int j = 1; j <= n; ++j) { int current_mod = prefix[j] % k; if (seen.find(current_mod) != seen.end()) { int i = seen[current_mod]; if (i <= j - 2) { return true; } } seen[current_mod] = j; } return false; } }};
代码解释
特殊情况处理：当 k 为 0 时，遍历数组检查是否有连续的两个 0。如果有，返回 true，否则返回 false。 前缀和计算：计算前缀和数组 prefix，其中 prefix[i] 表示数组前 i 个元素的和。 模运算记录：使用一个哈希表 seen 记录每个前缀和对 k 取模后的值及其对应的索引。当发现相同的模值时，检查索引是否满足子数组长度至少为 2 的条件，如果满足，返回 true。 遍历检查：遍历前缀和数组，更新模值记录，直到找到符合条件的子数组或遍历结束。 这种方法的时间复杂度为 O(n)，适用于较大的数组长度。
转载地址：http://vnki.baihongyu.com/