Koala - 10기/코딩테스트 준비 스터디

[백준/JAVA] #3273 두 수의 합

알 수 없는 사용자 2023. 3. 13. 21:41

https://www.acmicpc.net/problem/3273

 

3273번: 두 수의 합

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

www.acmicpc.net

 

문제

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)

출력

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

 

아이디어(알고리즘)

배열이 주어져있고 타겟하는 숫자가 매번 바뀌므로 이진탐색으로 구현하는 것이 빠를것같다.

코드

 

package Koala_study.week_2;

import java.util.Arrays;
import java.util.Scanner;

public class s_3273 {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);

        int n = in.nextInt();
        int[] arr = new int[n];

        for (int i = 0; i < n; i++) {
            arr[i] = in.nextInt();
        }

        int x = in.nextInt();

        int ans = 0;

        Arrays.sort(arr);

        for (int i : arr) {
            ans += BSearch(arr, x - i);
        }
        System.out.println(ans / 2);

    }

    static int BSearch(int[] arr, int find) {
        int left = 0;
        int right = arr.length - 1;
        int mid;


        while (right >= left) {
            mid = (right + left) / 2;
            if (arr[mid] < find) left = mid + 1;
            else if (arr[mid] > find) right = mid - 1;
            else return 1;
        }
        return 0;

    }


}

설명

각 배열을 입력받은 후 한번 정렬하고, 이후 이진탐색 메소드를 만들어 ans에 타겟넘버를 두수의합 - 배열의 각 숫자로 만들어 순서쌍이므로 ans / 2로 출력하였습니다.