-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path3Sum
57 lines (48 loc) · 1.75 KB
/
3Sum
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
#
//Runtime: 47 ms, faster than 35.83% of Java online submissions for 3Sum.
//Memory Usage: 60.1 MB, less than 32.20% of Java online submissions for 3Sum.
//Two pointers
//Time: O(N * LogN + N * N); Space : O(N + LogN)
class Solution {
public List<List<Integer>> threeSum(int[] nums) {
int target = 0;
List<List<Integer>> output = new ArrayList<>();
if(nums.length < 3) return output;
Arrays.sort(nums);
Set<List<Integer>> s = new HashSet<>();
for (int i = 0; i < nums.length; i++){
int j = i + 1;
int k = nums.length - 1;
while (j < k) {
int sum = nums[i] + nums[j] + nums[k];
if (sum == target) {
s.add(Arrays.asList(nums[i], nums[j], nums[k]));
j++;
k--;
} else if (sum < target) {
j++;
} else {
k--;
}
}
}
output.addAll(s);
return output;
}
}
------------------
Brute Force:
//TLE
//Time: O(N * N * N * log3); Space: O(N)
public List<List<Integer>> threeSum_brute(int[] nums) {
Set<List<Integer>> resultSet = new HashSet();
for (int i = 0; i < nums.length - 2; i++)
for (int j= i + 1; j < nums.length - 1; j++)
for (int k = j + 1 ; k < nums.length; k++)
if (0 == nums[i] + nums[j] + nums[k]) {
List<Integer> list = new ArrayList<>(Arrays.asList(nums[i], nums[j], nums[k]));
Collections.sort(list);
resultSet.add(list);
}
return new ArrayList<>(resultSet);
}