[백준] 2230번 수 고르기 (Python)

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

2230번: 수 고르기

 

처음에 정렬로만 풀었다가 시간초과가 났다.

이후 두 수를 고른다는 조건을 파악하여 투 포인터로 풀었다.

위 문제들이랑 결이 비슷해서 어렵지 않게 풀 수 있었다.

1.  입력받은 숫자들을 빈 배열에 추가해 준다.
2. 배열을 정렬한다.
3. 인덱스를 늘리며 두 수의 차를 구한다.
   
   • 두 수의 차가 m 미만인 경우, right 인덱스에 1을 더하여 더 큰 차이가 나도록 만든다.
   • 두 수의 차가 m 이상인 경우, 현재 m 이상인 최소의 차와 비교한다.
4. 결과를 출력한다.

 

정답 코드

import sys
input = sys.stdin.readline

n, m = map(int, input().split())
arr = []
for idx in range(n):
    arr.append(int(input()))

arr.sort()
answer = sys.maxsize
left = right = 0

while left <= right and right < n:
    diff = arr[right] - arr[left]
    
    if diff < m:
        right += 1
    
    else:
        left += 1
        answer = min(diff, answer)
    
print(answer)