기록

백준_2467_용액 본문

코딩테스트/python

백준_2467_용액

youngyin 2021. 1. 15. 19:30

문제

 

2467번: 용액

첫째 줄에는 전체 용액의 수 N이 입력된다. N은 2 이상 100,000 이하의 정수이다. 둘째 줄에는 용액의 특성값을 나타내는 N개의 정수가 빈칸을 사이에 두고 오름차순으로 입력되며, 이 수들은 모두 -

www.acmicpc.net

풀이

1. 구해야 하는 두 수 중 첫번째 수 a를 선택한다.

2. 지정한 수보다 오른쪽에 있는 수들을 탐색 범위[startPoint, endPoint]로 지정한다.

3. 이분 탐색을 통해 첫번째 수 a와의 합이 0에 가장 가까운 두번째 수 b를 찾는다.

  • 3-1. a+startPoint와 a+endPoint의 곱의 부호가 -이라면 startPoint는 startPoint와 endPoint의 중간이 된다.
  • 3-2. a+startPoint와 a+endPoint의 곱의 부호가 +이거나 곱이 0이라면 탐색을 중단한다.

4. a+startPoint와 a+endPoint 중 작은 값을 기록한다.

코드

N = int(input())
arr = list(map(int, input().split()))

num = abs(sum(arr[-2:])) # arr의 뒤에서 두자리 수를 선택
ans = arr[-2:] # arr의 뒤에서 두자리 수를 선택

for i in range(N-1) :
    s, e = i+1, N-1
    while e-s>1 :
        n1 = arr[s]+arr[i]
        n2 = arr[e]+arr[i]
        
        if n1*n2<0 :
            p = int((s+e)/2) # s와 e의 중간 지점
            n3 = arr[p]+arr[i]
            
            if n2*n3<=0 : s = p
            else : e = p
            
        # n1과 n2의 부호가 같거나 0이 두 수에 포함될 때
        else: break
	
    # 시작점과 끝점만 탐색
    for j in [s, e] :
        tmp = abs(arr[i]+arr[j])
        if num>tmp : num, ans = tmp, [arr[i], arr[j]]
print(*ans)

'코딩테스트 > python' 카테고리의 다른 글

백준_1654_랜선 자르기  (0) 2021.01.18
백준_1920_수 찾기  (0) 2021.01.18
백준_1806_부분합  (0) 2021.01.15
백준_11660_구간 합 구하기 5  (0) 2021.01.06
백준_1629_곱셈  (0) 2021.01.05
Comments