사용자 도구

사이트 도구


ps:problems:boj:2230

수 고르기

ps
링크acmicpc.net/…
출처BOJ
문제 번호2230
문제명수 고르기
레벨골드 5
분류

투 포인터

시간복잡도O(nlogn)
인풋사이즈n<=100,000
사용한 언어Python
제출기록33052KB / 164ms
최고기록136ms
해결날짜2021/12/07

풀이

  • 차이가 X에 가장 가까운 쌍을 구하는 문제.
  • 정렬된 리스트에서, 이진 검색 (Binary search)이나 투 포인터를 사용해서 풀 수 있다. 투 포인터를 사용하면 O(n), 이진 탐색을 사용하면 O(nlogn)이지만, 어차피 처음에 배열을 정렬하는데에 O(nlogn)이 걸리기 때문에 총 시간복잡도는 O(nlogn)이 된다.

코드

"""Solution code for "BOJ 2230. 수 고르기".

- Problem link: https://www.acmicpc.net/problem/2230
- Solution link: http://www.teferi.net/ps/problems/boj/2230

Tags: [Two Pointer]
"""

import sys

INF = float('inf')


def main():
    N, M = [int(x) for x in sys.stdin.readline().split()]
    nums = [int(sys.stdin.readline()) for _ in range(N)]

    nums.sort()
    min_diff = INF
    l_iter, r_iter = iter(nums), iter(nums)
    left, right = next(l_iter), next(r_iter)
    while right is not None:
        diff = right - left
        if M <= diff < min_diff:
            min_diff = diff
        if diff > M:
            left = next(l_iter)
        else:
            right = next(r_iter, None)

    print(min_diff)


if __name__ == '__main__':
    main()

토론

댓글을 입력하세요:
T W​ B K C
 
ps/problems/boj/2230.txt · 마지막으로 수정됨: 2021/12/08 02:05 저자 teferi