ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • [BOJ/Python] 백준 20922 - 겹치는 건 싫어
    Solve 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문 두개에서 터졌나