[백준 11440] 피보나치 수의 제곱의 합 [C]

백준 - 11440

---

풀이

n번째 피보나치 수의 제곱의 합은 다음과 같은 규칙으로 구할 수 있다.

위키백과 - 피보나치 수

"백준 11444, 피보나치 수 6" 코드를 사용해 풀이했다.

---

소스코드

#include <stdio.h>
#define ll long long
const int MOD = 1e9 + 7;
ll X[2][2] = {0, 1, 1, 1};

void fibo(ll result[][2], ll f[][2]){
    int temp[2][2] = {0,};
    for (int i = 0; i < 2; i++)
        for (int j = 0; j < 2; j++)
            for (int k = 0; k < 2; k++)
                temp[i][j] += (result[i][k]*f[k][j]) %MOD;

    for (int i = 0; i < 2; i++)
        for (int j = 0; j < 2; j++)
            result[i][j] = temp[i][j] %MOD;	
}
int main(){
    ll n1, result = 1;
    scanf("%lld", &n1);
    ll n = n1+1;
    for (int i = 0; i < 2; i++){
        ll temp[2][2] = {1, 0, 0, 1};
        while (n){
            if (n & 1) fibo(temp, X);
            fibo(X, X);
            n >>= 1;
        }
        result = result*temp[1][0] %MOD;
        X[0][0] = 0;
        X[0][1] = X[1][0] = X[1][1] = 1;
        n = n1;
    }
    printf("%lld", result);
}

---

출처 및 참고자료

11440번: 피보나치 수의 제곱의 합

[백준 11444] 피보나치 수 6 [C]

피보나치 수 - 위키백과, 우리 모두의 백과사전