[ProblemSolving] 백준 - 2470 두용액 (투포인터)

문제 링크

문제 설명

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

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

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

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

입력

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

출력

첫째 줄에 특성값이 0에 가장 가까운 용액을 만들어내는 두 용액의 특성값을 출력한다. 출력해야 하는 두 용액은 특성값의 오름차순으로 출력한다. 특성값이 0에 가장 가까운 용액을 만들어내는 경우가 두 개 이상일 경우에는 그 중 아무것이나 하나를 출력한다.

예제 입력1

5
-2 4 -99 -1 98

예제 출력1

-99 98

나의 풀이

투포인터으로 간단히 해결할 수 있는 문제다.

입력받은 용액을 정렬하고, (두 용액합의 절댓값, 용액1, 용액2)을 담을 리스트를 생성한다.

이 리스트에는 투포인터를 통해 비교를 하여 두 용액의 합의 절댓값이 작은 것을 담을 수 있도록 했다.

투포인터는 left와 right가 같은 용액을 가리킬 때까지 진행하며, 0에 더 가까운 용액이 발견되면, 리스트의 첫번째 요소를 변경해준다.

나의 코드

import sys
input = sys.stdin.readline
n = int(input())
arr = list(map(int, input().split())) 
arr.sort()
left, right = 0,n-1
result = []
result.append(( abs(arr[left] + arr[right]), arr[left] ,arr[right]))
while left <right:
    
    if arr[left] + arr[right] <=0:
        left += 1
        if left != right and result[0][0] > abs(arr[left] + arr[right]) :
            result[0]=(abs(arr[left] + arr[right]), arr[left] ,arr[right])

    else:
        right -= 1
        if left != right and result[0][0]  > abs(arr[left] + arr[right]) :
            result[0]=(abs(arr[left] + arr[right]), arr[left] ,arr[right])

print(result[0][1], result[0][2])

다른 분 풀이 2

투포인터을 이용한 풀이다. 나의 풀이보다 조금 더 빠르다.

다른 점은 투포인터 순서와 리스트에 꼭 필요한 두 용액만 표시해서 간단한 풀이를 제공했다.

다른 분 코드

import sys
INF = sys.maxsize
input = sys.stdin.readline

def sol():
  n = int(input())
  target = list(map(int, input().split()))
  target.sort()

  lt = 0
  rt = n - 1
  minN = INF

  res = []

  while lt < rt:
    curr = target[lt] + target[rt]
    if abs(curr) < minN:
      minN = abs(curr)
      res = [target[lt], target[rt]]
    if curr >= 0:
      rt -= 1
    else :
      lt += 1

  print(*res) # 컨테이너 타입의 데이터 언패킹(리스트, 튜플, 딕에서)

sol()