-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.cpp
More file actions
78 lines (72 loc) · 2.12 KB
/
main.cpp
File metadata and controls
78 lines (72 loc) · 2.12 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
// Source: https://leetcode.com/problems/continuous-subarray-sum
// Title: Continuous Subarray Sum
// Difficulty: Medium
// Author: Mu Yang <http://muyang.pro>
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 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`.
//
// **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
// ```
//
// **Constraints:**
//
// - `1 <= nums.length <= 10^5`
// - `0 <= nums[i] <= 10^9`
// - `0 <= sum(nums[i]) <= 2^31 - 1`
// - `1 <= k <= 2^31 - 1`
//
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
#include <unordered_set>
#include <vector>
using namespace std;
// Prefix sum
//
// Compute the prefix sum.
// If there are two prefix sum that are congruent modulo k,
// then good subarray exists.
class Solution {
public:
bool checkSubarraySum(vector<int>& nums, int k) {
int n = nums.size();
auto modMap = unordered_set<int>();
auto currSum = 0; // empty sum
for (auto num : nums) {
auto nextSum = (currSum + num) % k;
if (modMap.contains(nextSum)) return true;
modMap.insert(currSum);
currSum = nextSum;
}
return false;
}
};