[JAVA] 백준 3273 - 두 수의 합

문제

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

 

입력

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

 

출력

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

 

입출력 예제

 

알고리즘❓ 풀어내기❗️

1. 입력받은 숫자를 배열로 받는다.
2. 배열을 정렬한다.
3. 시작점 = 0, 끝점 = n - 1로 설정한 뒤 두 점에 위치한 원소끼리 더하며, 각 각 +1, -1 씩 인덱스를 옮겨간다.
4. 위의 과정을 거치며 두 원소의 합이 X와 같다면 count를 증가시켜준다.
5. 위의 내용은 투 포인터 알고리즘으로 설명할 수 있다. 이것에 대한 내용은 다음에 더 깊이 공부한 뒤 포스팅 하겠다.

 

🧑🏻‍💻코드 리뷰

import java.io.*;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        StringTokenizer st = new StringTokenizer(br.readLine());
        int[] arr = new int[N];
        for (int i = 0; i < N; i++)
            arr[i] = Integer.parseInt(st.nextToken());

        Arrays.sort(arr);

        int X = Integer.parseInt(br.readLine());
        int start = 0;
        int end = N - 1;
        int sum = 0;
        int count = 0;

        while (start < end) {
            sum = arr[start] + arr[end];
            if (sum < X) {
                start++;
            } else if (sum > X) {
                end--;
            } else {
                count++;
                start++;
                end--;
            }
        }
        System.out.println(count);
    }
}