Skip to content

Commit 95ce67e

Browse files
committed
add 3sum solution
1 parent 706f738 commit 95ce67e

File tree

1 file changed

+63
-0
lines changed

1 file changed

+63
-0
lines changed

3sum/Tessa1217.java

Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
import java.util.List;
2+
import java.util.ArrayList;
3+
import java.util.Arrays;
4+
5+
// 정수 배열 nums가 주어질 때 nums[i], nums[j], nums[k]로 이루어진 배열을 반환하시오
6+
// 반환 배열 조건: i 가 j와 같지 않고, i가 k와 같지 않으며 세 요소의 합이 0인 배열
7+
class Solution {
8+
9+
// 시간복잡도: O(n^2)
10+
public List<List<Integer>> threeSum(int[] nums) {
11+
12+
Arrays.sort(nums);
13+
14+
List<List<Integer>> answer = new ArrayList<>();
15+
16+
int left = 0;
17+
int right = 0;
18+
int sum = 0;
19+
20+
for (int i = 0; i < nums.length - 2 && nums[i] <= 0; i++) {
21+
22+
// 중복 제거
23+
if (i > 0 && nums[i] == nums[i - 1]) {
24+
continue;
25+
}
26+
27+
left = i + 1;
28+
right = nums.length - 1;
29+
30+
while (left < right) {
31+
32+
sum = nums[i] + nums[left] + nums[right];
33+
// System.out.println(String.format("%d, %d, %d", i, left, right));
34+
// System.out.println(String.format("%d + %d + %d = %d", nums[i], nums[left], nums[right], sum));
35+
if (sum < 0) {
36+
left++;
37+
continue;
38+
}
39+
if (sum > 0) {
40+
right--;
41+
continue;
42+
}
43+
44+
answer.add(List.of(nums[i], nums[left], nums[right]));
45+
46+
// 중복 제거
47+
while (left < right && left + 1 < nums.length && nums[left] == nums[left + 1]) {
48+
left++;
49+
}
50+
while (left < right && right - 1 >= 0 && nums[right] == nums[right - 1]) {
51+
right--;
52+
}
53+
left++;
54+
right--;
55+
56+
}
57+
}
58+
59+
return answer;
60+
}
61+
62+
}
63+

0 commit comments

Comments
 (0)