투포인터 문제 - 합이 0이 되는 숫자들 고르기 (투포인터로 중복 방지)

// 주어진 nums 배열에서 3 개의 합이 0이 되는 모든 숫자들을 출력하세요.
// 중복된 숫자 셋은 제외하고 출력하세요.

// 입출력 예시
// nums: {-1, 0, 1, 2, -1, -4}; 
// 출력: [[-1, -1, 2], [-1, 0, 1]]

// nums: {1, -7, 6, 3, 5, 2}
// 출력: [[-7, 1, 6], [-7, 2, 5]]

 

/**
 * 어려움 !
 */
public class Main {
    public static ArrayList<ArrayList<Integer>> solution(int[] nums) {
        Arrays.sort(nums);

        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        for (int i = 0; i < nums.length - 2; i++) {
            if (i != 0 && nums[i - 1] == nums[i]) {
                continue;
            }
            int p1 = i + 1;
            int p2 = nums.length - 1;

            while (p1 < p2) { //// -4 -1 -1 0 1 2
                if (nums[i] + nums[p1] + nums[p2] == 0) { // 리스트 담기
                    list.add(new ArrayList<>(Arrays.asList(nums[i], nums[p1], nums[p2])));
                    while (p1 < p2 && nums[p1] == nums[p1 + 1]) { // 중복 방지 -> nums[p1] == nums[p1 + 1]이면 같은 로직 발생 !
                        p1++;
                    }
                    while (p1 < p2 && nums[p2 - 1] == nums[p2]) { // 중복 방지
                        p2--;
                    }
                    p1++;
                    p2--;
                } else if (nums[i] + nums[p1] + nums[p2] > 0) {
                    p2--;
                } else { //nums[i] + nums[p1] + nums[p2] < 0
                    p1++;
                }
            }
        }

        return list;
    }

    public static void main(String[] args) {
        // Test code

        // -4 -1 -1 0 1 2
        int[] nums = {-1, 0, 1, 2, -1, -4};     // [[-1, -1, 2], [-1, 0, 1]]
        System.out.println(solution(nums));

        nums = new int[]{1, -7, 6, 3, 5, 2};   // [[-7, 1, 6], [-7, 2, 5]]
        System.out.println(solution(nums));
    }
}