[acmicpc] 20922. 겹치는 건 싫어(python)

이중 for문으로 구현하면 터져서 O(n)으로 구현했다

import sys
input=sys.stdin.readline

N,K=map(int,input().split())
arr=list(map(int,input().split()))

left,right=0,0
check=[0]*100001
answer=0
cnt=0
while right<N:
    if check[arr[right]]<K:
        check[arr[right]]+=1
        cnt+=1
        right += 1
    else:
        answer=max(answer,cnt)
        cnt-=1
        check[arr[left]]-=1
        left+=1
answer=max(answer,cnt)
print(answer)

---

https://www.acmicpc.net/problem/20922

20922번: 겹치는 건 싫어