2022.02.25 62일차

백준 2470번 (두 용액)

문제

KOI 부설 과학연구소에서는 많은 종류의 산성 용액과 알칼리성 용액을 보유하고 있다. 각 용액에는 그 용액의 특성을 나타내는 하나의 정수가 주어져있다. 산성 용액의 특성값은 1부터 1,000,000,000까지의 양의 정수로 나타내고, 알칼리성 용액의 특성값은 -1부터 -1,000,000,000까지의 음의 정수로 나타낸다.

같은 양의 두 용액을 혼합한 용액의 특성값은 혼합에 사용된 각 용액의 특성값의 합으로 정의한다. 이 연구소에서는 같은 양의 두 용액을 혼합하여 특성값이 0에 가장 가까운 용액을 만들려고 한다.

예를 들어, 주어진 용액들의 특성값이 [-2, 4, -99, -1, 98]인 경우에는 특성값이 -99인 용액과 특성값이 98인 용액을 혼합하면 특성값이 -1인 용액을 만들 수 있고, 이 용액이 특성값이 0에 가장 가까운 용액이다. 참고로, 두 종류의 알칼리성 용액만으로나 혹은 두 종류의 산성 용액만으로 특성값이 0에 가장 가까운 혼합 용액을 만드는 경우도 존재할 수 있다.

산성 용액과 알칼리성 용액의 특성값이 주어졌을 때, 이 중 두 개의 서로 다른 용액을 혼합하여 특성값이 0에 가장 가까운 용액을 만들어내는 두 용액을 찾는 프로그램을 작성하시오.

나의 풀이

1. 0에 가까운 두 수를 구하기, 시간 복잡도 고려 - 투 포인터, 오름차순 정렬
2. 음수 or 양수 경우 고려하기 위해 절댓값(Math.abs)
3. start, end값 설정, while문 반복
4. 절대값의 두 합이 최소값보다 작으면 최소값 교체, 두 정답 값 교체
5. 절대값의 두 합이 0보다 크면 end값 감소
6. 절대값의 두 합이 0보다 작으면 start값 증가

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

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

        int n = Integer.parseInt(st.nextToken());

        long[] nArr = new long[n];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            long temp = Integer.parseInt(st.nextToken());
            nArr[i] = temp;
        }

        Arrays.sort(nArr);

        long answer1 = 0;
        long answer2 = 0;

        int start = 0;
        int end = n - 1;

        long min = Integer.MAX_VALUE;

        while (start < end) {
            long temp = Math.abs(nArr[end] + nArr[start]);
            if (nArr[end] + nArr[start] > 0) {
                if (min > temp) {
                    min = temp;
                    answer1 = nArr[start];
                    answer2 = nArr[end];
                }
                end--;

            } else {
                if (min > temp) {
                    min = temp;
                    answer1 = nArr[start];
                    answer2 = nArr[end];
                }
                start++;
            }
        }
        System.out.print(answer1 + " " + answer2);

    }
}

생각하기

• 최소값 비교 코드 중복, 먼저 따로 비교해도 상관 없음

 while (start < end) {
            long temp = Math.abs(nArr[end] + nArr[start]);
            if (min > temp) {
                min = temp;
                answer1 = nArr[start]; answer2 = nArr[end];
            }
            if (nArr[end] + nArr[start] > 0) {
                end--;
            } else {
                start++;
            }
        }