[백준] #3273 두수의 합 - 파이썬

문제

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

입력

첫째 줄에 수열의 크기 n이 주어진다. 다음 줄에는 수열에 포함되는 수가 주어진다. 셋째 줄에는 x가 주어진다. (1 ≤ n ≤ 100000, 1 ≤ x ≤ 2000000)

출력

문제의 조건을 만족하는 쌍의 개수를 출력한다.

예제 입력 1 복사

9
5 12 7 10 9 1 2 3 11
13

예제 출력 1 복사

3

---

1. combination으로 풀었더니 시간 초과가 났다. 오토에버는 이런 류의 문제가 많나보다.

2. 투포인터로 풀었더니 쉽게 풀 수 있었따. 

---

솔루션

import sys
from itertools import combinations

n = int(sys.stdin.readline())
ls = list(map(int, sys.stdin.readline().split()))
x = int(sys.stdin.readline())

left, right = 0, n-1
ls.sort()

cnt = 0

while left<right:
    cur_sum = ls[left]+ls[right]
    if cur_sum == x:
        cnt += 1
        left += 1
        right -= 1
    elif cur_sum > x:
        right -= 1
    else:
        left += 1

print(cnt)