[Python] 백준 3273번: 투 포인터 사용

📎 백준 3273번 풀러 가기!

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)

출력

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

---

오늘은 투 포인터 문제다!

 

투 포인터는 두 개의 포인터를 사용하여 문제를 해결하는 방법이다.

 

오늘 푼 문제는 투 포인터의 아주 대표적인 문제이기 때문에

머릿속으로 동작 방식을 생각하면서 구현했더니 쉽게 풀 수 있었다.

 

이렇게 오늘도 문제 하나 완료!

---

아래는 내가 구현한 코드👇

import sys

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

cnt = 0

i = 0
j = len(num) - 1

num.sort()

while i < j:
    temp = num[i] + num[j]
    if temp == x:
        cnt += 1
        i += 1
        j -= 1
    elif temp < x:
        i += 1
    else:
        j -= 1

print(cnt)