[백준|실버3] 3273번 두 수의 합 - Python(파이썬)

문제

https://www.acmicpc.net/problem/3273

✓ 시간 제한: 1초 / 메모리 제한: 128MB

 

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

---

첫 번째 풀이 - 시간초과

이중 반복문을 이용했다. 시간초과 날거라고 예상은 했지만 .. 🥲

import sys
input = sys.stdin.readline

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

x = int(input())
res = 0

for i in range(0, arr_len):
    for j in range(i+1, arr_len+1):
        if i+j == x:
            res += 1

print(res)

---

두 번째 풀이 - 투 포인터 이용

투 포인터를 이용해 풀이한다.

 

left는 배열의 왼쪽 인덱스, right는 배열은 오른쪽 인덱스이고,

두 수의 합과 x 값을 비교하여 좌우 인덱스를 변경한다.

 

먼저 수가 들어있는 배열을 오름차순으로 정렬하고 탐색을 진행한다.

1. 두 수의 합이 x와 같은 경우: 결과값 +1, left +1

2. 두 수의 합이 x보다 작은 경우: left +1

3. 두 수의 합이 x보다 큰 경우: right -1

 

탐색은 left와 right가 만날 때까지 진행한다.

 

예제 입력을 시뮬레이션 돌리면 아래와 같다.

이렇게 한 번의 배열 탐색으로 경우를 확인할 수 있다.

import sys
input = sys.stdin.readline

arr_len = int(input())
arr = sorted(list(map(int, input().split())))
x = int(input())

res = 0
left, right = 0, arr_len-1

while left < right:
    temp = arr[left] + arr[right]
    if temp == x:
        res += 1
        left += 1
    elif temp < x:
        left += 1
    else:
        right -= 1

print(res)