Skip to content

Commit 5e2c8eb

Browse files
committed
Added 4sum
1 parent c997a78 commit 5e2c8eb

File tree

4 files changed

+137
-0
lines changed

4 files changed

+137
-0
lines changed
Lines changed: 38 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,38 @@
1+
#include <vector>
2+
#include <algorithm>
3+
4+
using namespace std;
5+
6+
class Solution {
7+
public:
8+
vector<vector<int>> fourSum(vector<int>& nums, int target) {
9+
vector<vector<int>> answer;
10+
sort(nums.begin(), nums.end());
11+
int n = nums.size();
12+
13+
for (int i = 0; i < n; i++) {
14+
if (i > 0 && nums[i] == nums[i - 1]) continue;
15+
16+
for (int j = i + 1; j < n; j++) {
17+
if (j > i + 1 && nums[j] == nums[j - 1]) continue;
18+
19+
int lo = j + 1, hi = n - 1;
20+
while (lo < hi) {
21+
long long sum = (long long)nums[i] + nums[j] + nums[lo] + nums[hi];
22+
if (sum == target) {
23+
answer.push_back({nums[i], nums[j], nums[lo], nums[hi]});
24+
lo++;
25+
hi--;
26+
while (lo < hi && nums[lo] == nums[lo - 1]) lo++;
27+
while (lo < hi && nums[hi] == nums[hi + 1]) hi--;
28+
} else if (sum < target) {
29+
lo++;
30+
} else {
31+
hi--;
32+
}
33+
}
34+
}
35+
}
36+
return answer;
37+
}
38+
};
Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
import java.util.ArrayList;
2+
import java.util.Arrays;
3+
import java.util.List;
4+
5+
public class Solution {
6+
public List<List<Integer>> fourSum(int[] nums, int target) {
7+
List<List<Integer>> answer = new ArrayList<>();
8+
Arrays.sort(nums);
9+
int n = nums.length;
10+
11+
for (int i = 0; i < n; i++) {
12+
if (i > 0 && nums[i] == nums[i - 1]) continue;
13+
14+
for (int j = i + 1; j < n; j++) {
15+
if (j > i + 1 && nums[j] == nums[j - 1]) continue;
16+
17+
int lo = j + 1, hi = n - 1;
18+
while (lo < hi) {
19+
long sum = (long) nums[i] + nums[j] + nums[lo] + nums[hi];
20+
if (sum == target) {
21+
answer.add(Arrays.asList(nums[i], nums[j], nums[lo], nums[hi]));
22+
lo++;
23+
hi--;
24+
while (lo < hi && nums[lo] == nums[lo - 1]) lo++;
25+
while (lo < hi && nums[hi] == nums[hi + 1]) hi--;
26+
} else if (sum < target) {
27+
lo++;
28+
} else {
29+
hi--;
30+
}
31+
}
32+
}
33+
}
34+
return answer;
35+
}
36+
}
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
var fourSum = function(nums, target) {
2+
let answer = [];
3+
nums.sort((a, b) => a - b);
4+
let n = nums.length;
5+
6+
for (let i = 0; i < n; i++) {
7+
if (i > 0 && nums[i] === nums[i - 1]) continue;
8+
9+
for (let j = i + 1; j < n; j++) {
10+
if (j > i + 1 && nums[j] === nums[j - 1]) continue;
11+
12+
let lo = j + 1, hi = n - 1;
13+
while (lo < hi) {
14+
let sum = nums[i] + nums[j] + nums[lo] + nums[hi];
15+
if (sum === target) {
16+
answer.push([nums[i], nums[j], nums[lo], nums[hi]]);
17+
lo++;
18+
hi--;
19+
while (lo < hi && nums[lo] === nums[lo - 1]) lo++;
20+
while (lo < hi && nums[hi] === nums[hi + 1]) hi--;
21+
} else if (sum < target) {
22+
lo++;
23+
} else {
24+
hi--;
25+
}
26+
}
27+
}
28+
}
29+
return answer;
30+
};
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
class Solution:
2+
def fourSum(self, nums: List[int], target: int) -> List[List[int]]:
3+
n = len(nums)
4+
answer = []
5+
nums.sort()
6+
7+
for i in range(n):
8+
if i > 0 and nums[i] == nums[i-1]:
9+
continue
10+
11+
for j in range(i+1, n):
12+
if j > i + 1 and nums[j] == nums[j-1]:
13+
continue
14+
15+
lo, hi = j + 1, n - 1
16+
while lo < hi:
17+
summ = nums[i] + nums[j] + nums[lo] + nums[hi]
18+
if summ == target:
19+
answer.append([nums[i], nums[j], nums[lo], nums[hi]])
20+
lo += 1
21+
hi -= 1
22+
while lo < hi and nums[lo] == nums[lo - 1]:
23+
lo += 1
24+
while lo < hi and nums[hi] == nums[hi + 1]:
25+
hi -= 1
26+
elif summ < target:
27+
lo += 1
28+
else:
29+
hi -= 1
30+
31+
return answer
32+
# Time: O(n^3)
33+
# Space: O(n)

0 commit comments

Comments
 (0)