[Python] 백준 2470 두 용액

2470번: 두 용액

투포인터

 

1) 용액 배열을 오름차순으로 정렬한다.

2) sp는 제일 왼쪽 인덱스 출발, ep는 제일 오른쪽 인덱스 출발

3) sp가 ep보다 커지거나 서로 만나면 while문 종료

4) 합쳐질 용액의 값 두개와 합의 절댓값을 계속 저장하고, 합이 > 0 이면 ep를 -1하고, 합이 < 0이면 sp를 +1 한다.

-> 0보다 크면 합을 더 작게 만들어야 하기에 오른쪽 포인터를 왼쪽으로 한칸 옮기고 

-> 0보다 작으면 합을 더 크게 만들어야 하기에 왼쪽 포인터를 오른쪽으로 한칸 옮긴다.

-> 0이면 바로 탈출

5) 절댓값을 기준으로 정렬하고, 합쳐졌던 용액의 값 두개를 출력한다.

 

전체 코드

import sys
input = sys.stdin.readline

N = int(input())
yongaeks = list(map(int, input().split()))
yongaeks.sort()

sp = 0
ep = len(yongaeks) - 1
sum_yongaeks = []
while sp < ep:
    sum = yongaeks[sp] + yongaeks[ep]
    tmp = [yongaeks[sp],yongaeks[ep],abs(sum)]
    sum_yongaeks.append(tmp)
    if sum > 0:
        ep -= 1
    elif sum < 0:
        sp += 1
    else:
        break

sum_yongaeks.sort(key= lambda x : x[2])
print(sum_yongaeks[0][0], sum_yongaeks[0][1])