523. Continuous Subarray Sum

1. Description

Given an integer array nums and an integer k, return true if nums has a good subarray or false otherwise.
A good subarray is a subarray where:

  • its length is at least two, and
  • the sum of the elements of the subarray is a multiple of k.

Note that:

  • A subarray is a contiguous part of the array.
  • An integer x is a multiple of k if there exists an integer n such that x = n * k. 0 is always a multiple of k.

2. Example

Example 1:
Input: nums = [23,2,4,6,7], k = 6
Output: true
Explanation: [2, 4] is a continuous subarray of size 2 whose elements sum up to 6.

Example 2:
Input: nums = [23,2,6,4,7], k = 6
Output: true
Explanation: [23, 2, 6, 4, 7] is an continuous subarray of size 5 whose elements sum up to 42. 42 is a multiple of 6 because 42 = 7 * 6 and 7 is an integer.

Example 3:
Input: nums = [23,2,6,4,7], k = 13
Output: false

3. Constraints

  • 1 <= nums.length <= $10^5$
  • 0 <= nums[i] <= $10^9$
  • 0 <= sum(nums[i]) <= $2^{31} - 1$
  • 1 <= k <= $2^{31} - 1$

4. Solutions

Prefix Sum & Hash Table

n = nums.size()
Time complexity: O(n)
Space complexity: O(min(n, k))

// if a % k == b % k, then a - b % k == 0
class Solution {
public:
    bool checkSubarraySum(const vector<int> &nums, int k) {
        map<int, int> reminder_index;
        reminder_index[0] = -1; // to calculate the subarray length when the index is 0
        for (int i = 0, reminder = 0; i < nums.size(); ++i) {
            reminder = (reminder + nums[i]) % k;
            if (reminder_index.count(reminder)) {
                if (i - reminder_index[reminder] > 1) {
                    return true;
                }
            } else {
                reminder_index[reminder] = i;
            }
        }

        return false;
    }
};
comments powered by Disqus