HIHO_矩阵快速幂

最新推荐文章于 2020-12-23 13:18:16 发布

梦里江山

最新推荐文章于 2020-12-23 13:18:16 发布

阅读量290

点赞数

分类专栏： ACM水题刷耍

本文链接：https://blog.csdn.net/whj_lim/article/details/45112109

版权

ACM水题刷耍专栏收录该内容

63 篇文章 0 订阅

订阅专栏

时间限制:10000ms
单点时限:1000ms
内存限制:256MB

描述

骨牌，一种古老的玩具。今天我们要研究的是骨牌的覆盖问题：
我们有一个2xN的长条形棋盘，然后用1x2的骨牌去覆盖整个棋盘。对于这个棋盘，一共有多少种不同的覆盖方法呢？
举个例子，对于长度为1到3的棋盘，我们有下面几种覆盖方式：

[week41_1.PNG]

提示：骨牌覆盖

提示：如何快速计算结果
输入

第1行：1个整数N。表示棋盘长度。1≤N≤100,000,000
输出

第1行：1个整数，表示覆盖方案数 MOD 19999997
样例输入

62247088

样例输出

17748018

#include<iostream>
#include<cstdio>
#include<cstring>
#define Mod 19999997
using namespace std;
struct matrix
{
    long long x[3][3];
};
matrix cheng(matrix a, matrix b)
{
    matrix c;
    for(int i = 1; i <= 2; i++)
    {
        for(int j = 1; j <= 2; j++)
        {
            c.x[i][j] = 0;
            for(int k = 1; k <= 2; k++)
            {
                 c.x[i][j] = (c.x[i][j] + a.x[i][k]*b.x[k][j])%Mod;
                // c.x[i][j] = (c.x[i][j]%Mod + a.x[i][k]%Mod*b.x[k][j]%Mod)%Mod;
            }
        }
    }
    return c;
}
matrix ksm(matrix a, int k)
{
    matrix e;
    e.x[1][2] = e.x[2][1] = 0;
    e.x[1][1] = e.x[2][2] = 1;
    while(k)
    {
        if(k&1)
        {
            e = cheng(e,a);
        }
        k>>=1;
        a = cheng(a,a);
    }
    return e;
}
void solve(int n)
{
    matrix e,c;
    e.x[1][2] = e.x[2][1] = e.x[2][2] = 1;
    e.x[1][1] = 0;
    e = ksm(e,n);
    c.x[1][1] = c.x[1][2] = 1;
    c.x[2][1] = c.x[2][2] = 0;
    c = cheng(c,e);
    printf("%lld\n",c.x[1][2]);
}
int main()
{
    int n;
    while(scanf("%d",&n) != EOF)
    {
        solve(n-1);
    }
    return 0;
}