Solve

[BOJ/Python] 백준 20922 - 겹치는 건 싫어

헨헨7 2024. 9. 10. 10:37

문제

더보기

문제

홍대병에 걸린 도현이는 겹치는 것을 매우 싫어한다. 특히 수열에서 같은 원소가 여러 개 들어 있는 수열을 싫어한다. 도현이를 위해 같은 원소가 $K$개 이하로 들어 있는 최장 연속 부분 수열의 길이를 구하려고 한다.

$100\,000$ 이하의 양의 정수로 이루어진 길이가 $N$인 수열이 주어진다.  이 수열에서 같은 정수를 $K$개 이하로 포함한 최장 연속 부분 수열의 길이를 구하는 프로그램을 작성해보자.

입력

첫째 줄에 정수 $N$ ($1 \le N \le 200\,000$)과 $K$ ($1 \le K \le 100$)가 주어진다.

둘째 줄에는 ${a_1, a_2, ... a_n}$이 주어진다 ($1 \le a_i \le 100\,000$)

출력

조건을 만족하는 최장 연속 부분 수열의 길이를 출력한다.

풀이

from collections import Counter

n, k = map(int, input().split())
li = list(map(int, input().split()))

left = 0
right = 0
ans = 1
arr = []

while right < n - 1:
    right += 1
    arr.append(li[right])
    # print(max(Counter(arr).values()))
    while max(Counter(arr).values()) > k:
        arr.pop(left)
        left += 1
        ans = max(ans, right - left + 1)
print(ans)
  1. 포인터의 위치를 left, right 모두 0으로 두고, ans를 1로 초기화한다.
  2. right 포인터를 한 칸씩 옮기면서 arr에 해당 값을 add한다.
  3. Counter 함수를 통해 빈도수를 구해 k를 초과할 경우, arr에서 left번째 값을 제외하고 left 포인터를 한 칸 이동한다.
  4. ans와 포인터 간의 간격을 비교하며 정답을 구한다.

근데 런타임 에러가 났습니다.. while문 두개에서 터졌나