백준 3273번 풀이 - 파이썬(Python) (정렬 / 두 포인터)

난이도: 실버3

https://www.acmicpc.net/problem/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(input())
array = sorted(list(map(int, sys.stdin.readline().split())))	# 리스트 입력 시 정렬하여 저장
x = int(input())

count = 0

for i in range(len(array)):
    for j in range(i, len(array)):
        if array[i] >= (x//2 + 1):	# 앞의 수와 더했을 때 x보다 반드시 큰 경우 제외
            break
        if array[i] + array[j] == x:	# 이미 탐색한 경우(i와 i-1 등)를 제외하여 i, j 설정
            count += 1
	
print(count)

알고리즘은 맞는 것 같은데

시간초과로 답이 나오지 않는다..

남의 풀이가 궁금하다.

 

 

 

 

남의 풀이 (두 포인터 사용)

 

import sys

n = int(input())
array = sorted(list(map(int, sys.stdin.readline().split())))	# 리스트를 정렬하여 입력 받기
x = int(input())

count = 0
left, right = 0, n-1	# 두 포인터 설정

while left < right:	# 두 포인터가 교차되는 순간 반복문 종료
    sum = array[left] + array[right]
    if sum == x:	# left, right가 가리키는 값의 합이 x와 같은 경우
        count += 1
        left += 1	# left를 이동시켜 sum > x로 만든다
    elif sum > x:	# 값이 큰 경우 right를 이동시켜 값을 줄인다
        right -= 1
    elif sum < x:	# 값이 작은 경우 left를 이동시켜 값을 키운다
        left += 1
    
print(count)

시간 복잡도가 크게 줄어 시간 초과 없이 문제를 풀 수 있었다.

'나의 풀이' 에서 사용한 알고리즘은, 중첩 반복문이기에 O(N^2)의 시간 복잡도를 가진다.

 

'남의 풀이' 에서 사용한 알고리즘은, 좌우 포인터가 움직이면서 N번의 탐색을 하기에

O(N)의 시간 복잡도를 가진다.

 

두 포인터에 대해 알게 된 문제였다.