ABC115 C - Christmas Eve
備忘録
問題
回答
import sys import os import math import string ii = lambda: int(sys.stdin.buffer.readline().rstrip()) il = lambda: list(map(int, sys.stdin.buffer.readline().split())) fl = lambda: list(map(float, sys.stdin.buffer.readline().split())) iln = lambda n: [int(sys.stdin.buffer.readline().rstrip()) for _ in range(n)] iss = lambda: sys.stdin.buffer.readline().decode().rstrip() sl = lambda: list(map(str, sys.stdin.buffer.readline().decode().split())) isn = lambda n: [sys.stdin.buffer.readline().decode().rstrip() for _ in range(n)] lcm = lambda x, y: x * y / math.gcd(x, y) MOD = 10 ** 9 + 7 MAX = float('inf') def main(): if os.getenv("LOCAL"): sys.stdin = open("input.txt", "r") N, K = il() H = [ii() for _ in range(N)] H = sorted(H, reverse=True) cnt = len(H)-K+1 ret = [0]*cnt for i in range(cnt): ret[i] = H[i] - H[i+K-1] print(min(ret)) if __name__ == '__main__': main()
考え方
木の高さH
をソートし、
高い順から1
番目と1 + K
番目の差分、
2
番目と2 + K
番目の差分 ... N-K
番目とN + K
番目の差分を求める。
飾り付けるK
本の木は最も高い木と最も低い木の差分が小さいければ、
間の木を飛ばすことが出来る。
つまり木の植えられた位置は順不同で考えることができるため、
ソートを行って先頭から順にK
本の木の高さの差分を求めることで、
最小の差を探索する。