알고리즘 공부/이진탐색 | 삼진탐색(그이상)

[백준] 2467 용액

kdhoooon 2021. 8. 22. 22:48

문제


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

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

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

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

 

 

 

풀이


오름차순으로 정렬이 된 상태로 입력이 주어지기 때문에 따로 정렬을 할 필요는 없다.

 

두개의 용액을 더한다에서 이분탐색으로 풀어야겠다고 생각을 했고, 이분탐색으로 두 수의 합의 최소를 찾았다.

int left = 0;
int right = n - 1;
long min = liquid[left] + liquid[right];
int a = left, b = right;
while(left < right) {
	long sum = liquid[left] + liquid[right];
	
	if(Math.abs(min) >= Math.abs(sum)) {
		min = sum;
		a = left;
		b = right;
	}
	
	if( sum < 0 ) {
		left++;
	}
	else {
		right--;
	}
}

 

만약 합이 0 보다 작으면 음수가 큰것이므로 left++ 을 해주고

합이 양수면 양수가 큰것이므로 right++ 을 해주는 방식으로 두개의 수를 더해준 절댓값의 최소를 구해주면 된다.

 

 

 

<전체코드>

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

public class Main {	

	static StringBuilder sb = new StringBuilder();
	static int[] liquid;
	
		
	static int parseInt(String string) {
		return Integer.parseInt(string);
	}
	
	public static void main(String[] args) throws IOException{
		BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
		
		int n = parseInt(bufferedReader.readLine());
		liquid = new int[n];
		
		StringTokenizer st = new StringTokenizer(bufferedReader.readLine());
		for(int i = 0 ; i < n ; i++) {
			liquid[i] = parseInt(st.nextToken());
		}
		
		int left = 0;
		int right = n - 1;
		long min = liquid[left] + liquid[right];
		int a = left, b = right;
		while(left < right) {
			long sum = liquid[left] + liquid[right];
			
			if(Math.abs(min) >= Math.abs(sum)) {
				min = sum;
				a = left;
				b = right;
			}
			
			if( sum < 0 ) {
				left++;
			}
			else {
				right--;
			}
		}
		
		System.out.println(liquid[a] + " " + liquid[b]);
	}
}