🔙뒤로가기

알고리즘 분류

문제

n개의 서로 다른 양의 정수 a1, a2, ..., an으로 이루어진 수열이 있다. ai의 값은 1보다 크거나 같고, 1000000보다 작거나 같은 자연수이다. 자연수 x가 주어졌을 때, ai + aj = x (1 ≤ i < j ≤ n)을 만족하는 (ai, aj)쌍의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 수열의 크기 n이 주어진다. 다음 줄에는 수열에 포함되는 수가 주어진다. 셋째 줄에는 x가 주어진다. (1 ≤ n ≤ 100000, 1 ≤ x ≤ 2000000)

출력

문제의 조건을 만족하는 쌍의 개수를 출력한다.

예제 입력 1

9
5 12 7 10 9 1 2 3 11
13

예제 출력 1

3

풀이

import java.util.*;

public class Main {
    public static void main(String[] args) throws Exception {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] arr = new int[n];
        for(int i = 0; i < arr.length; i++) {
            arr[i] = sc.nextInt();
        }
        int X = sc.nextInt();

        // 정렬 후 투 포인터 알고리즘을 사용한다.
        Arrays.sort(arr);
        int l = 0, r = arr.length - 1, count = 0;
        // 두 포인터가 만나면 종료한다.
        while(l < r) {
            // 합이 X와 같다면 count를 증가시키고 우측 포인터를 좌측으로 보낸다.
            if(arr[l] + arr[r] == X) {
                count++;
                l++;
                r--;
            } else if (arr[l] + arr[r] < X){
                // 합이 X보다 크면 왼쪽 포인터를 증가시킨다.
                l++;
            } else {
                // 합이 X보다 작다면 우측 포인터를 감소시킨다.
                r--;
            }
        }
        System.out.println(count);
    }
}

키포인트