[JAVA]두 수의 합(Baekjoon)

문제

문제

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

입력

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

출력

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

 

예제 입력 1

9
5 12 7 10 9 1 2 3 11
13

예제 출력 1

3

 

풀이

입력의 크기가 1,000,000이하이다. $O(nlogn)$까지 가능하다. 사실 유사한 문제를 많이 풀어봤던 기억 덕에 정렬 후, 이중 포인터로 접근해서 문제를 풀어야겠다고 바로 떠올랐다. 특정 값을 구하기 위한 두 원소 합의 조합 문제이다.

 

코드

import java.util.*;
import java.io.*;
public class Main{
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());  // 원소의 수
        int[] arr = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        for(int i = 0; i < N; i++)
            arr[i] = Integer.parseInt(st.nextToken());
        int target = Integer.parseInt(br.readLine()); // 타겟
        int answer = 0, lo = 0, hi = N-1; // 이중 포인터
        Arrays.sort(arr); // 정렬
        // i와 j가 달라야하기에 같은 경우는 포함되지 않는 케이스이다.
        // 두 원소의 합과 타켓 값을 비교해 포인터를 이동해가며 정답 케이스 탐색
        while(hi > lo){
            int mid = arr[lo] + arr[hi];
            if(mid < target){
                lo++;
            }
            else{
                if(mid == target)
                    answer++;
                hi--;
            }
        }
        System.out.print(answer);
    }
}