Skip to content

Commit 9939032

Browse files
committed
2022-06-21 update: added "3Sum"
1 parent 2af7635 commit 9939032

File tree

2 files changed

+73
-0
lines changed

2 files changed

+73
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,54 @@
1+
package com.smlnskgmail.jaman.leetcodejava.medium;
2+
3+
import java.util.ArrayList;
4+
import java.util.Arrays;
5+
import java.util.List;
6+
7+
// https://leetcode.com/problems/3sum/
8+
public class ThreeSum {
9+
10+
private final int[] input;
11+
12+
public ThreeSum(int[] input) {
13+
this.input = input;
14+
}
15+
16+
public List<List<Integer>> solution() {
17+
Arrays.sort(input);
18+
List<List<Integer>> result = new ArrayList<>();
19+
for (int i = 0; i < input.length - 2; i++) {
20+
int curr = input[i];
21+
if (i > 0 && curr == input[i - 1]) {
22+
continue;
23+
}
24+
int rem = -curr;
25+
int start = i + 1;
26+
int end = input.length - 1;
27+
while (start < end) {
28+
int s = input[start];
29+
int e = input[end];
30+
if (s + e == rem) {
31+
List<Integer> part = new ArrayList<>();
32+
part.add(curr);
33+
part.add(s);
34+
part.add(e);
35+
result.add(part);
36+
while (start < end && input[start] == input[start + 1]) {
37+
start++;
38+
}
39+
while (start < end && input[end] == input[end - 1]) {
40+
end--;
41+
}
42+
start++;
43+
end--;
44+
} else if (s + e > rem) {
45+
end--;
46+
} else {
47+
start++;
48+
}
49+
}
50+
}
51+
return result;
52+
}
53+
54+
}
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
package com.smlnskgmail.jaman.leetcodejava.medium;
2+
3+
import org.junit.Test;
4+
5+
import java.util.List;
6+
7+
import static org.junit.Assert.assertEquals;
8+
9+
public class ThreeSumTest {
10+
11+
@Test
12+
public void defaultTest() {
13+
assertEquals(
14+
List.of(List.of(-1, -1, 2), List.of(-1, 0, 1)),
15+
new ThreeSum(new int[]{-1, 0, 1, 2, -1, -4}).solution()
16+
);
17+
}
18+
19+
}

0 commit comments

Comments
 (0)