Scribbling

[Java 101] LeetCode: 15. 3Sum with 2D List 본문

Computer Science/Java

[Java 101] LeetCode: 15. 3Sum with 2D List

focalpoint 2023. 2. 14. 01:15
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        Arrays.sort(nums);
        int N = nums.length;
        for (int i = 0; i < N - 2; i++) {
            if (i > 0 && nums[i] == nums[i-1]) {
                continue;
            }
            int j = i + 1, k = N - 1;
            while (j < k) {
                int n1 = nums[i], n2 = nums[j], n3 = nums[k];
                int sum = n1 + n2 + n3;
                if (sum == 0) {
                    List<Integer> temp = List.of(n1, n2, n3);
                    ret.add(temp);
                    while (j < k && nums[j] == n2) {
                        j += 1;
                    }
                    while (j < k && nums[k] == n3) {
                         k -= 1;
                    }
                } else if (sum < 0) {
                    j += 1;
                } else {
                    k -= 1;
                }
            }
        }
        return ret;
    }
}