-
Notifications
You must be signed in to change notification settings - Fork 229
/
Copy path1590-make-sum-divisible-by-p.js
121 lines (103 loc) · 2.88 KB
/
1590-make-sum-divisible-by-p.js
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
/**
* @param {number[]} nums
* @param {number} p
* @return {number}
*/
var minSubarray = function(nums, p) {
const n = nums.length
const sum = nums.reduce((acc, cur) => acc + cur, 0)
const target = sum % p
if(target === 0) return 0
const hash = {0: -1}
let res = n
let curSum = 0
for(let i = 0; i < n; i++) {
const e = nums[i]
curSum += e
const remain = curSum % p
const diff = (remain - target + p) % p
if(hash[diff] != null) {
res = Math.min(res, i - hash[diff])
}
hash[remain] = i
}
return res === n ? -1 : res
};
// another
/**
* @param {number[]} nums
* @param {number} p
* @return {number}
*/
const minSubarray = function(nums, p) {
const sum = nums.reduce((ac, e) => ac+ e,0)
const target = sum % p, n = nums.length, {min} = Math
if(target === 0) return 0
const map = new Map()
map.set(0, -1)
let res = n
for(let i = 0, s = 0; i < n; i++) {
s += nums[i]
const r = s % p
if(r >= target) {
if(map.has(r - target)) res = min(res, i - map.get(r-target))
}else {
if(map.has(p + r - target)) res = min(res, i - map.get(p + r - target))
}
map.set(r, i)
}
return res === n ? -1 : res
};
// another
/**
* @param {number[]} nums
* @param {number} p
* @return {number}
*/
const minSubarray = function(nums, p) {
const remain = nums.reduce((ac, e) => ac + e, 0) % p
const n = nums.length, hash = {0: -1}
let res = n
if(remain === 0) return 0
for(let i = 0, sum = 0; i < n; i++) {
const cur = nums[i]
sum += cur
const target = (sum % p - remain + p) % p
if(hash[target] != null) {
res = Math.min(res, i - hash[target])
}
hash[sum % p] = i
}
// console.log(hash)
return res === n ? -1 : res
};
// another
/**
* @param {number[]} nums
* @param {number} p
* @return {number}
*/
const minSubarray = function(nums, p) {
const diff = nums.reduce((a, b) => a + b, 0) % p;
let res = diff === 0 ? 0 : nums.length;
for (let i = 0, sum = 0, map = {0: -1}; i < nums.length; i++) {
sum += nums[i];
const target = (sum % p - diff + p) % p;
if (map[target] !== undefined) {
res = Math.min(res, i - map[target]);
}
map[sum % p] = i;
}
return res === nums.length ? -1 : res;
};
/**
Let pre[] be the prefix sum array,
then pre[i] is running prefix sum or prefix sum of i elements,
pre[j] is the prefix sum such that pre[i]-pre[j] is the subarray we
need to remove to make pre[n] (sum of all elements) divisible by p
(pre[n] - (pre[i]-pre[j])) % p = 0 ... (remove a subarray to make pre[n] divisible by p)
=> pre[n] % p = (pre[i]-pre[j]) % p ... ((a-b)%m = a%m - b%m)
=> pre[j]%p = pre[i]%p - pre[n]%p ... (same property used above)
since RHS can be negative we make it positive modulus by adding p and taking modulus
=> pre[j]%p = (pre[i]%p - pre[n]%p + p) % p
*/