[백준] 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

풀이

투 포인터를 이용한 문제이다.
투 포인터 문제는 대부분 정렬을 필요로 한다. (아닌 경우도 있음)

입력 배열의 처음과 마지막 위치를 start, end 포인터로 설정하고,

• 두 수의 합이 x 보다 큰 경우 -> start 를 뒤로 한 단계 이동
• 두 수의 합이 x 보다 작은 경우 -> end 를 앞으로 한 단계 이동
• 두 수의 합이 x 인 경우 -> start, end 를 각각 앞, 뒤로 한 단계 이동

이 과정을 start < end 인 조건 동안 반복해주면 된다.
문제의 조건이 start < end 이기 때문이다.

import sys
input = sys.stdin.readline

n = int(input())
arr = list(map(int,input().split()))
arr.sort()
x = int(input())

s, e = 0, len(arr)-1
cnt = 0

while s < e:
    sum_val = arr[s] + arr[e]
    if sum_val > x:
        e  -= 1
    elif sum_val < x:
        s += 1
    else:
        s += 1
        e -= 1
        cnt += 1
print(cnt)