-
[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)
- 포인터의 위치를 left, right 모두 0으로 두고, ans를 1로 초기화한다.
- right 포인터를 한 칸씩 옮기면서 arr에 해당 값을 add한다.
- Counter 함수를 통해 빈도수를 구해 k를 초과할 경우, arr에서 left번째 값을 제외하고 left 포인터를 한 칸 이동한다.
- ans와 포인터 간의 간격을 비교하며 정답을 구한다.
근데 런타임 에러가 났습니다.. while문 두개에서 터졌나
'Solve' 카테고리의 다른 글
[BOJ/Python] 백준 1548 - 부분 삼각 수열 (0) 2024.09.20 [BOJ/Python] 백준 2470 - 두 용액 (0) 2024.09.10 [BOJ/Python] 백준 21921 - 블로그 (0) 2024.09.10 [BOJ/Python] 백준 2302 - 극장 좌석 (0) 2024.08.26 [BOJ/Python] 백준 1660 - 캡틴 이다솜 (0) 2024.08.26