[Python/파이썬] 백준 3273번 두 수의 합

3273번: 두 수의 합

 

코드

n = int(input())
a = sorted(list(map(int, input().split())))
x = int(input())

ans = 0
for i in range(n):
    s, e = i+1, n-1
    while s <= e:
        mid = (s+e)//2
        if a[i]+a[mid] == x:
            ans += 1
            break
        elif a[i]+a[mid] < x:
            s = mid+1
        else:
            e = mid-1
print(ans)

 

$a_i$와 더해서 $x$가 되는 특정 수 $a_j$를 찾는 것이기 때문에 배열 $a$를 오름차순 정렬해놓고 이분탐색으로 $a_j$를 찾았다.