BOJ 1253 좋다

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

문제

$N$개의 수 중에서 어떤 수가 다른 수 두 개의 합으로 나타낼 수 있다면 그 수를 “좋다(GOOD)”고 한다.

$N$개의 수가 주어지면 그 중에서 좋은 수의 개수는 몇 개인지 출력하라.

수의 위치가 다르면 값이 같아도 다른 수이다.

---

입력

첫째 줄에는 수의 개수 $N(1 ≤ N ≤ 2,000)$, 두 번째 줄에는 $i$번째 수를 나타내는 $A_i$가 $N$개 주어진다. $(|A_i| ≤ 1,000,000,000, A_i$는 정수$)$

---

출력

좋은 수의 개수를 첫 번째 줄에 출력한다.

---

예제 입력 1

10
1 2 3 4 5 6 7 8 9 10

예제 출력 1

8

---

힌트

3,4,5,6,7,8,9,10은 좋다.

---

코드

투 포인터로 문제를 풀고 나서 문제의 알고리즘 분류를 보니, 이분 탐색도 있었다.

다음에 문제를 풀게 되면 이분 탐색으로도 풀어보자!

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {

    static int N;
    static int[] numbers;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        numbers = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            numbers[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(numbers);

        int answer = 0;
        for (int i = 0; i < N; i++) {
            if(isGood(i)) {
                answer++;
            }
        }

        System.out.println(answer);
    }

    static boolean isGood(int idx) {
        int lIdx = 0;
        int rIdx = N - 1;

        while (lIdx < rIdx) {
            // lIdx나 rIdx가 검사 중인 수의 인덱스(idx)와 같은 경우, 건너 뛴다.
            if (lIdx == idx) {
                lIdx++;
            } else if (rIdx == idx) {
                rIdx--;
            } else {
                // 두 수의 합
                int sum = numbers[lIdx] + numbers[rIdx];

                // 두 수의 합과 같으면 좋은 수이다.
                if(sum == numbers[idx]) {
                    return true;
                }

                // 두 수의 합보다 현재 검사 중인 수가 작으면 rIdx--, 크면 lIdx++
                if(sum > numbers[idx]) {
                    rIdx--;
                } else {
                    lIdx++;
                }
            }
        }

        return false;
    }
}