백준 2470번 두 용액

Solution

import java.io.*;
import java.util.*;

public class Main {
    static FastReader scan = new FastReader();
    static StringBuilder sb = new StringBuilder();

    static int N;
    static int[] A;

    static void input() {
        N = scan.nextInt();
        A = new int[N + 1];
        for (int i = 1; i <= N; i++) {
            A[i] = scan.nextInt();
        }
    }

    static int lower_bound(int[] A, int L, int R, int X) {
        // A[L...R] 에서 X 이상의 수 중 제일 왼쪽 인덱스를 return 하는 함수
        // 그런 게 없다면 R + 1 을 return
        int result = R + 1;

        while (L <= R) {
            int mid = (L + R) / 2;
            // 찾고자 하는 수의 조건
            if (A[mid] >= X) {
                // 위치 갱신
                result = mid;
                // 더 작은 걸 찾아야되니까 오른쪽 땡겨옴, 모르겠으면 그림그리기
                R = mid - 1;
            } else {
                L = mid + 1;
            }
        }
        return result;
    }

    static void pro() {

        Arrays.sort(A, 1, N + 1);

        int best_sum = Integer.MAX_VALUE;
        int v1 = 0, v2 = 0;
        for (int left = 1; left <= N - 1; left++) {
            // -A[left] 와 가장 가까운 용액을 자신의 오른쪽 구간에서 찾자
            int candidate = lower_bound(A, left + 1, N, -A[left]);

            if (candidate == N + 1) {
                candidate--;
            }

            // 왼쪽도 확인해줘야 함, -1 할 때는 항상 범위 조심하자
            if (candidate - 1 > left && Math.abs(A[left] + A[candidate]) >= Math.abs(A[left] + A[candidate - 1])) {
                candidate--;
            }

            int newSum = Math.abs(A[left] + A[candidate]);
            if (newSum <= best_sum) {
                best_sum = newSum;
                v1 = A[left];
                v2 = A[candidate];
            }
        }
        sb.append(v1).append(' ').append(v2);
        System.out.println(sb);
    }

    public static void main(String[] args) {
        input();
        pro();
    }


    static class FastReader {
        BufferedReader br;
        StringTokenizer st;

        public FastReader() {
            br = new BufferedReader(new InputStreamReader(System.in));
        }

        public FastReader(String s) throws FileNotFoundException {
            br = new BufferedReader(new FileReader(new File(s)));
        }

        String next() {
            while (st == null || !st.hasMoreElements()) {
                try {
                    st = new StringTokenizer(br.readLine());
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            return st.nextToken();
        }

        int nextInt() {
            return Integer.parseInt(next());
        }

        long nextLong() {
            return Long.parseLong(next());
        }

        double nextDouble() {
            return Double.parseDouble(next());
        }

        String nextLine() {
            String str = "";
            try {
                str = br.readLine();
            } catch (IOException e) {
                e.printStackTrace();
            }
            return str;
        }
    }
}