글 읽기 - 반례가 궁금합니다

pycubic0915 1년 전 1

도저히 반례를 못찾겠습니다....

고수님들 반례 부탁드리겠습니다.

import sys

input = sys.stdin.readline

# 행렬을 최댓값까지 계산
dp = [0, 1, 2]
for i in range(3, 41):
    dp.append(dp[i - 1] + dp[i - 2])

# 좌석 수
N = int(input())
arr = [0] * N
# 입력 받는 횟수
M = int(input())
for _ in range(M):
    arr[int(input()) - 1] = 1

# 계산
ans = 0
chair = 0
for i in range(N):
    # vip 좌석일 경우
    if arr[i] == 1:
        if chair != 0:
            # 답에 0을 곱하면 0이기 때문에 1 더하기
            if ans == 0:
                ans += 1
            ans = ans * dp[chair]
        chair = 0
        
    # 마지막 좌석이 vip 좌석이 아닌 경우
    elif i == N - 1:
        chair += 1
        # 답에 0을 곱하면 0이기 때문에 1 더하기
        if ans == 0:
            ans += 1
        ans = ans * dp[chair]
    else:
        chair += 1

print(ans)