-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path4sum2.js
64 lines (54 loc) · 1.61 KB
/
4sum2.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
/* 4Sum II
Given four integer arrays nums1, nums2, nums3, and nums4 all of length n,
return the number of tuples (i, j, k, l) such that:
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
Example 1:
Input: nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
Output: 2
Explanation:
The two tuples are:
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
Example 2:
Input: nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
Output: 1
Constraints:
n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228
*/
/**
* @param {number[]} nums1
* @param {number[]} nums2
* @param {number[]} nums3
* @param {number[]} nums4
* @return {number}
*/
var fourSumCount = function(nums1, nums2, nums3, nums4) {
let tuples = 0;
let n = nums1.length;
let tuplesSum = {};
for(let i = 0; i < n; i++){
for(let j = 0; j < n ; j++){
let sum = nums1[i] + nums2[j];
if(sum in tuplesSum){
tuplesSum[sum]++;
}else{
tuplesSum[sum] = 1;
}
}
}
for(let i = 0; i < n; i++){
for(let j = 0; j < n ; j++){
let sum = -1 * (nums3[i] + nums4[j]);
if(tuplesSum[sum]){
tuples += tuplesSum[sum];
}
}
}
return tuples;
};