35708670번 코드가 다음과 같은 입력 데이터에서 시간 초과가 발생합니다.

데이터가 너무 약하거나 문제가 잘못되거나 둘 중 하나의 문제가 있는 것 같습니다.

해당 코드는 0ms로 통과했지만, N=1000인 경우 K=50이 아니라 K=20만 되어도 런타임이 2초에 육박합니다.
문제 설명에 적힌 K 범위가 잘못 되었거나, 데이터가 너무 약한 것 같습니다.

+ 추가

assert 문을 사용 해보니 모든 데이터가 N <= 20, K <= 20 입니다. 문제 데이터를 새로 만들거나 문제 조건을 완전히 바꿔야 할 것 같습니다.