백준 2531. 회전 초밥

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

2531번: 회전 초밥

n, d, k, c = map(int, input().split()) # 접시수/초밥가짓수/연속해서먹는접시수/쿠폰번호
arr = []
ans = 0
for _ in range(n):
    num = int(input())
    arr.append(num)

for i in range(n):
        if i+k > n:  # 한바퀴 돌아?
            check = len(set(arr[i:n] + arr[:(i+k)%n] + [c]))
        else:
            check = len(set(arr[i:i+k]+[c]))
        if ans < check:
            ans = check
print(ans)

- set으로 묶어 len으로 종류의 갯수를 구하는 방법이 있다는 것에 깨달음을 얻었다.

- 처음에 짠 코드가 시간초과가 나서 다른 코드를 참고했다...

- 종류의 갯수를 셀 때 쿠폰인 c를 무조건 포함시켜야됨!