본문 바로가기
Python/이것이 취업을 위한 코딩 테스트다(with 파이썬)

이것이 취업을 위한 코딩테스트다 - 바닥공사

by sophia02 2022. 7. 17.
# 바닥 공사
n = int(input())
array = []
array.append(1)  # 첫째 항을 배열에 저장
array.append(3)  # 두번째 항을 배열에 저장
if n == 1:
    print(1 % 796796)  # 1일경우
elif n == 2:
    print(3 % 796796)  # 2일경우
else:
    for i in range(n-2):  # 3 이상일 경우
        array.append(2*array[i]+array[i+1])
    print(array[-1] % 796796)