[Sliding Window] 백준 2531번 “회전 초밥” Python 풀이

문제 링크

• https://www.acmicpc.net/problem/2531

풀이

• 슬라이딩 윈도우 알고리즘과 딕셔너리를 사용해서 해결할 수 있습니다.
  • 윈도우의 크기를 k로 설정한 뒤,
  • 윈도우를 오른쪽으로 한 칸씩 이동하며,
  • 구간 내에 존재하는 접시의 종류별 개수를 파악하면 됩니다.
• 참고
  • 회전 초밥 벨트는 원형으로 이루어져있기 때문에, 오른쪽 인덱스는 오른쪽 인덱스를 접시의 수로 나눈 나머지를 사용하면 됩니다.
  • 구간 내에는 항상 쿠폰 번호가 포함되어 있다고 가정합니다.

코드

import sys
from collections import defaultdict

# n : 회전 초밥 벨트에 놓인 접시의 수
# d : 초밥의 가짓수
# k : 연속해서 먹는 접시의 수
# c : 쿠폰 번호
n, d, k, c = map(int, sys.stdin.readline().split())

# 초밥 접시 상황
arr = []
for _ in range(n):
    arr.append(int(sys.stdin.readline()))

# 구간 인덱스 초기화
left, right = 0, k-1

# 구간 내의 접시 종류별 개수
dict = defaultdict(int)

# 구간 내에는 항상 쿠폰 번호가 포함되어있다고 가정
dict[c] += 1

# 첫 시작 구간의 접시 종류별 개수 저장
for i in range(right+1):
    dict[arr[i]] += 1

# 구간 내의 최대 접시 종류 개수 초기화
result = -int(1e9)

# 슬라이딩 윈도우 진행
while left < n:

    # 현재 구간 내의 접시 종류 개수와 최대 접시 종류 개수를 비교
    result = max(len(dict), result)

    # 윈도우를 오른쪽으로 한 칸씩 이동하기 위한 작업 진행

    # 현재 구간 내의 가장 왼쪽 접시를 제거
    dict[arr[left]] -= 1
    # 이제 해당 종류의 접시가 남아있지 않다면, 딕셔너리에서 제거
    if (dict[arr[left]] == 0):
        del dict[arr[left]]
    # 왼쪽 인덱스를 오른쪽으로 한 칸 이동
    left += 1

    # 오른쪽 인덱스를 오른쪽으로 한 칸 이동
    right += 1
    # 현재 구간 내의 가장 오른쪽 접시를 추가
    dict[arr[right % n]] += 1

# 결과 출력
print(result)