Lang:Python2
Edit1234567891011121314151617181920212223from collections import Counterdef solve(a, N, K):b = Counter(a)sk = sorted(b.keys())for i in range(len(sk)):k, v = sk[i], b[sk[i]]if v > N / K:return Falseif v == 0:continuefor j in range(1, K):kk = k + jif b[kk] < v:return Falseb[kk] -= vreturn Truefor c in range(input()):N, K = map(int, raw_input().split())if N % K:print 'NO'continuea = map(int, raw_input().split())print 'YES' if solve(a, N, K) else 'NO'