Algorithm/BOJ

세 용액 - 2473

jhg0406 2020. 4. 16. 13:08
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
//2473 - 세 용액
 
#include <iostream>
#include <algorithm>
using namespace std;
 
int N;
long long ans;
long long arr[5000];
long long target;
int idx1, idx2, idx3;
 
void init()
{
    ans = 3000000000
    cin >> N;
    for (int i = 0; i < N; ++i)
        cin >> arr[i];
    sort(arr, arr + N);
}
 
int BS(int left, int right)
{
    if(left == right) return left;
 
    int mid = (left + right)/2;
    if(target <= arr[mid])
        return BS(left, mid);
    else
        return BS(mid+1, right);
}
 
long long absol(long long a)
{
    return (a > 0 ? a : -a);
}
 
int find(int idx)
{
    if(idx == N-1return idx;
    long long lgap = absol(target - arr[idx]);
    long long rgap = absol(target - arr[idx+1]);
    if(lgap < rgap) return idx;
    return idx+1;
}
 
int main()
{
    ios_base::sync_with_stdio(0), cin.tie(0);
    init();
    for (int i = 0; i < N - 2++i)
        for(int j = i+1; j<N-1++j)
        {
            target = -(arr[i] + arr[j]);
            int k = BS(j+1, N-1);
            if(target < arr[k])
            {
                if(k-1 != j)
                    k = find(k-1);
            }
            else
                k = find(k);
            long long sum = arr[i] + arr[j] + arr[k];
            if(absol(sum) < ans)
            {
                ans = absol(sum);
                idx1 = i, idx2 = j, idx3 = k;
            }
        }
    cout << arr[idx1] << " " << arr[idx2] << " " << arr[idx3];
}
cs

 

 

 

 

 

 

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

 

2473번: 세 용액

첫째 줄에는 전체 용액의 수 N이 입력된다. N은 3 이상 5,000 이하의 정수이다. 둘째 줄에는 용액의 특성값을 나타내는 N개의 정수가 빈칸을 사이에 두고 주어진다. 이 수들은 모두 -1,000,000,000 이상 1,000,000,000 이하이다. N개의 용액들의 특성값은 모두 다르고, 산성 용액만으로나 알칼리성 용액만으로 입력이 주어지는 경우도 있을 수 있다.

www.acmicpc.net

 

 

 

 

 

 

세 용액

두 개의 용액을 먼저 선택합니다. 두 용액의  특성값을 더하고, 그 합의 부호와 반대되는 값(target)과 가장 비슷한 값을 나머지 영역에서 이분탐색으로 찾아주어 세 개의 용액의 절대값이 가장 작은것을 찾아주면 됩니다.

주의할 점은 이분탐색으로 찾은 idx를 가지는 용액의 특성값이 target보다 크다면 idx-1, idx의 용액중 더 가까운것을 선택해야하고, target이 크다면 idx, idx+1 용액중 더 가까운것을 선택해야 합니다.