poj3070 - Fibonacci(矩阵快速幂）

最新推荐文章于 2022-04-19 20:43:25 发布

Tw_light

最新推荐文章于 2022-04-19 20:43:25 发布

阅读量202

点赞数

本文链接：https://blog.csdn.net/Tw_light/article/details/104031156

版权

刷题专栏收录该内容

18 篇文章 0 订阅

订阅专栏

题目：http://poj.org/problem?id=3070
题目大意：

$\begin{pmatrix} 1 & 1 \\ 1 & 0 \end{pmatrix} ^n的值$

解题方法：矩阵快速幂
code：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
const int N = 2;
const int mod = 10000;
int res[N][N];
int temp[N][N];
int num[N][N];
void multi(int a[][N], int b[][N])
{
    memset(temp, 0, sizeof(temp));
    for(int i = 0; i < N; i++)
        for(int j = 0; j < N; j++)
        for(int k = 0; k < N; k++)
        temp[i][j] = (temp[i][j] + a[i][k] * b[k][j]) % mod;

    for(int i = 0; i < N; i++)
        for(int j = 0; j < N; j++)
        a[i][j] = temp[i][j];
}
void MatPow(int a[][N], int n)
{
    memset(res, 0, sizeof(res));
    for(int i = 0; i < N; i++) res[i][i] = 1;

    while(n)
    {
        if(n&1)
        {
            multi(res, a);
        }
        multi(a, a);
        n >>= 1;
    }
}
int main()
{
    int n;
    while(~scanf("%d", &n) && (n != -1))
    {
        num[0][0] = 1;
        num[0][1] = 1;
        num[1][0] = 1;
        num[1][1] = 0;

        if(n == 0 || n == 1) printf("%d\n", n);
        else
        {
            MatPow(num, n);
            printf("%d\n", res[0][1]);
        }
    }
    return 0;
}