[백준 2473] 세 용액 C++

문제

백준 2473 세 용액 C++

2473번: 세 용액

풀이

용액문제 (https://gamedoridori.tistory.com/238)의 심화형이다.

오름차순으로 용액이 정렬되어있지 않으므로 먼저 정렬을 해준다.

그리고 각각의 용액에 대해, 그 용액과 합의 절댓값이 최소가 되는 두 용액을 두 포인터로 찾아준다.

 

여담으로 용액과 두 용액 문제는 입력값이 -10억~10억이라 int 자료형으로도 문제가 없었지만, 세 용액 문제는 int 자료형을 사용할 시에 abs 함수 사용시에 문제점이 있었다. 다른 해결책도 있겠지만, 편의상 long long 자료형으로 바꿔주었다.

소스 코드

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
 
    int N;
    cin >> N;
    vector<long long> input;
    input.reserve(N);
         
    for (int i = 0; i < N; i++)
    {
        long long num = 0;
        cin >> num;
        input.push_back(num);
    }
 
    sort(input.begin(), input.end());
 
    long long result1 = input[0];
    long long result2 = input[1];
    long long result3 = input[N - 1];
    long long minAdd = abs(result1 + result2 + result3);
 
    for (int idx = 0; idx < N; idx++)
    {
        int start = 0;
        int end = N - 1;
 
        while (start < end)
        {
    ㅅ        if (start == idx)
            {
                start++;
                continue;
            }
                
            else if (end == idx)
            {
                end--;
                continue;
            }
 
            long long add = input[start] + input[end] + input[idx];
            long long absAdd = std::abs(add);
            if (absAdd < minAdd)
            {
                minAdd = absAdd;
                result1 = input[start];
                result2 = input[end];
                result3 = input[idx];
            }
 
            if (add > 0)
            {
                end--;
            }
            else
            {
                start++;
            }
        }
    }
 
 
    vector<long long> result;
    result.push_back(result1);
    result.push_back(result2);
    result.push_back(result3);
 
    sort(result.begin(), result.end());
 
    for (auto e : result)
    {
        cout << e << ' ';
    }
 
    return 0;
}