[BAEKJOON] 3273번: 두 수의 합

문제

n개의 서로 다른 양의 정수 a1, a2, ..., an으로 이루어진 수열이 있다. ai의 값은 1보다 크거나 같고, 1000000보다 작거나 같은 자연수이다. 자연수 x가 주어졌을 때, ai + aj = x (1 ≤ i < j ≤ n)을 만족하는 (ai, aj)쌍의 수를 구하는 프로그램을 작성하시오.

 

코드

# 두 수의 합
n = input()
arr = set(map(int, input().split()))
x = int(input())
ans = 0
# set의 in 연산으로 문제 풀기
for i in arr:
    # set의 in 연산의 시간 복잡도는 O(1)이다.
    if x - i in arr:
        ans += 1

# 쌍의 개수를 출력해야 되니까 2로 나누어 준다.
print(ans // 2)

# 투 포인터를 이용해서 문제 풀기
# arr 리스트를 정렬 해줘야 된다.
ans = 0
left, right = 0, n - 1  # 왼쪽, 오른쪽
while left < right:
    temp = arr[left] + arr[right]
    if temp == x:
        ans += 1
        left += 1
    elif temp < x:
        left += 1
    else:
        right -= 1
print(ans)

 

나의 생각

set을 sorted()를 이용해서 정렬하면 리스트로 반환한다.
list의 in 연산은 시간복잡도가 O(N)이라서 시간 초과가 발생한다.