给你一个整数数组 nums
和一个整数 k
,如果 nums
有一个 好的子数组 返回 true
,否则返回 false
:
一个 好的子数组 是:
- 长度 至少为 2 ,且
- 子数组元素总和为
k
的倍数。
注意:
- 子数组 是数组中 连续 的部分。
- 如果存在一个整数
n
,令整数x
符合x = n * k
,则称x
是k
的一个倍数。0
始终 视为k
的一个倍数。
输入: nums = [23,2,4,6,7], k = 6 输出: true 解释: [2,4] 是一个大小为 2 的子数组,并且和为 6 。
输入: nums = [23,2,6,4,7], k = 6 输出: true 解释: [23, 2, 6, 4, 7] 是大小为 5 的子数组,并且和为 42 。 42 是 6 的倍数,因为 42 = 7 * 6 且 7 是一个整数。
输入: 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
use std::collections::HashMap;
impl Solution {
pub fn check_subarray_sum(nums: Vec<i32>, k: i32) -> bool {
let mut prefix_mod = 0;
let mut mod_indices = HashMap::from([(0, -1)]);
for i in 0..nums.len() {
prefix_mod = (prefix_mod + nums[i]) % k;
match mod_indices.get(&prefix_mod) {
Some(j) if j + 1 < i as i32 => return true,
None => {
mod_indices.insert(prefix_mod, i as i32);
}
_ => (),
}
}
false
}
}