문제
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]);
}
}
'알고리즘 공부 > 이진탐색 | 삼진탐색(그이상)' 카테고리의 다른 글
[리트코드] sliding-window-maximum (슬라이딩 윈도우) <Java> (0) | 2021.12.02 |
---|---|
[프로그래머스] 징검다리 건너기 (0) | 2021.05.11 |
백준 1644 (소수의 연속합)* (0) | 2021.04.24 |
백준 1208 (부분수열의 합2) (0) | 2021.04.23 |
백준 7453 (합이 0인 네 정수) (0) | 2021.04.22 |