출처 : 수 고르기
Solution
Python
import sys
input = sys.stdin.readline
if __name__ == '__main__':
N, M = map(int, input().split())
A = [int(input()) for _ in range(N)]
A.sort() # 일단 정렬해준다.
# 초기값은 M의 최대값으로 설정
answer = 2000000000
i, j = 0, 1 # i가 더 작은 수, j는 더 큰 수 (포인터)
# j가 i보다 언제나 크니까, j가 마지막 인덱스보다 커지면 탈출
while j < N:
# 두 수의 차이 n
n = A[j]-A[i]
# M보다 클 경우,
if n >= M:
# 현재까지 차이와 n 중 더 작은 값을 고름
answer = min(answer, n)
i += 1 # 이 때는 더 작은 포인터만 움직여준다.
# 다른 경우에는 더 큰 포인터를 움직여서 두 수 차이를 늘려줘야함(M보다 커지도록)
else: j += 1
# 언제나 j는 i보다 커야하니까, 만약 같아지면 j를 +1 해준다.
if i == j: j += 1
print(answer)
# 메소드 체이닝 기법으로 한번에 연결
# from itertools import combinations
# print(min(filter(lambda x: x >= M, map(lambda x: abs(x[0]-x[1]),combinations(A, 2)))))
문제
N개의 정수로 이루어진 수열 A[1], A[2], …, A[N]이 있다. 이 수열에서 두 수를 골랐을 때(같은 수일 수도 있다), 그 차이가 M 이상이면서 제일 작은 경우를 구하는 프로그램을 작성하시오.
예를 들어 수열이 {1, 2, 3, 4, 5}라고 하자. 만약 M = 3일 경우, 1 4, 1 5, 2 5를 골랐을 때 그 차이가 M 이상이 된다. 이 중에서 차이가 가장 작은 경우는 1 4나 2 5를 골랐을 때의 3이 된다.
입력
첫째 줄에 두 정수 N, M이 주어진다. 다음 N개의 줄에는 차례로 A[1], A[2], …, A[N]이 주어진다.
출력
첫째 줄에 M 이상이면서 가장 작은 차이를 출력한다. 항상 차이가 M이상인 두 수를 고를 수 있다.
제한
- 1 ≤ N ≤ 100,000
- 0 ≤ M ≤ 2,000,000,000
- 0 ≤ |A[i]| ≤ 1,000,000,000
예제 입력 1
3 3 1 5 3
예제 출력 1
4