523. 连续的子数组和
给你一个整数数组 nums 和一个整数 k ,编写一个函数来判断该数组是否含有同时满足下述条件的连续子数组:
子数组大小 至少为 2 ,且
子数组元素总和为 k 的倍数。
如果存在,返回 true ;否则,返回 false 。
如果存在一个整数 n ,令整数 x 符合 x = n * k ,则称 x 是 k 的一个倍数。0 始终视为 k 的一个倍数。
这道题最开始的想法是暴力求解,但是由于O(n2)的时间复杂度,导致超时
使用前缀和,哈希表remain2index中,key为前缀和除以k的余数,value为第一次出现的位置。因为一个整数除以另一个数和其余数除以该数的结果和余数都相等,而如果两个前缀和除以k的余数相等,那么这两个前缀和之差,显然是k的倍数,此时只需要这两个前缀和的位置相差大于等于2(保证是一个子数组),那么就可以返回true
需要额外注意的点是余数为零的情况,如果nums数组的第一个元素即为k的倍数,在以上判断中仍为true,但是不满足题目的要求,因此应该特殊判断
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
unordered_map<int, int> remain2index;// 前缀和除以k的余数 这个余数第一次出现的位置
int n = nums.size();
if