문제
https://www.acmicpc.net/problem/11444
11444번: 피보나치 수 6
첫째 줄에 n이 주어진다. n은 1,000,000,000,000,000,000보다 작거나 같은 자연수이다.
www.acmicpc.net
피보나치 수는 0과 1로 시작한다. 0번째 피보나치 수는 0이고, 1번째 피보나치 수는 1이다. 그 다음 2번째 부터는 바로 앞 두 피보나치 수의 합이 된다.
이를 식으로 써보면 Fn = Fn-1 + Fn-2 (n ≥ 2)가 된다.
n=17일때 까지 피보나치 수를 써보면 다음과 같다.
0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233, 377, 610, 987, 1597
n이 주어졌을 때, n번째 피보나치 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 n이 주어진다. n은 1,000,000,000,000,000,000보다 작거나 같은 자연수이다.
출력
첫째 줄에 n번째 피보나치 수를 1,000,000,007으로 나눈 나머지를 출력한다.
예제
입력 | 출력 |
1000 | 517691607 |
문제풀이 코드
import sys
input = sys.stdin.readline
N = int(input())
mod = 10**9 + 7
def mul(matrix1, matrix2):
res = [[0] * 2 for _ in range(2)]
for i in range(2):
for j in range(2):
for k in range(2):
res[i][j] += matrix1[i][k] * matrix2[k][j]
res[i][j] %= mod
return res
def devide(n, matrix):
if n == 1:
return matrix
else:
temp = devide(n//2, matrix)
if n % 2 == 0:
return mul(temp, temp)
else:
return mul(mul(temp, temp), matrix)
matrix = [[1, 1], [1, 0]]
if N - 1 < 2:
print(1)
else:
print(devide(N-1, matrix)[0][0])
풀이방식
1. 동적계획법으로 풀려고했으나, n의 최댓값이 엄청 크기때문에 동적계획법으로 해결할 수 없었다. 이에 행렬을 이용하여 피보나치 수를 구할 수 있다는 것을 확인하고 분할정복으로 행렬의 n제곱을 구하는 방식으로 해결하였다.
2. mul 함수는 matrix1과 matrix2를 곱해주는 함수이다.
3. devide함수는 n을 2로 계속 나누면서 행렬을 연산해주는 역할을 한다.
- n == 1이면 기본 행렬을 반환한다.
- n % 2 == 0 이면 temp의 제곱을 반환한다.
- n % 2 != 0 이면 temp의 제곱을 구하고 현재 matrix와의 곱셈을 구하여 반환한다.
4. devide함수를 실행하면서 temp의 값이 계속해서 연산이 되며 n제곱의 값을 얻을 수 있다.
5. devide를 시작할 때 matrix는 피보나치 수의 1에 해당하기 때문에 N에 -1을 해준 결과를 반환해준다.(-1을 안하고 [0][1]의 값을 출력해도 됨)
'Problem Solving > 백준' 카테고리의 다른 글
[백준] 12865.평범한 배낭 (0) | 2023.03.02 |
---|---|
[백준] 9251. LCS (0) | 2023.03.01 |
[백준] 9938.방 청소 (0) | 2023.01.20 |
[백준] 2021.최소 환승 경로 (0) | 2022.12.15 |
[백준] 1708.볼록 껍질 (0) | 2022.11.27 |