FZu 1683 纪念SlingShot 矩阵快速幂

最新推荐文章于 2024-09-11 15:01:00 发布

lenga5241

最新推荐文章于 2024-09-11 15:01:00 发布

阅读量571

点赞数

分类专栏：矩阵快速幂文章标签：矩阵

本文链接：https://blog.csdn.net/u011632342/article/details/40118375

版权

矩阵快速幂专栏收录该内容

2 篇文章 0 订阅

订阅专栏

题目链接：http://acm.fzu.edu.cn/problem.php?pid=1683

题目大意：已知 F（n）=3 * F（n-1）+2 * F（n-2）+7 * F（n-3），n>=3，其中F（0）=1，F（1）=3，F（2）=5，对于给定的每个n，输出F（0）+ F（1）+ …… + F（n） mod 2009。

解题思路：对于递推一类的问题，应该比较快的方法就是矩阵快速幂了，其中精华就是矩阵的构造。

假设前n项的和为S（n），则S（n）=S(n-1)+F(n)=S(n-1)+3 * F（n-1）+2 * F（n-2）+7 * F（n-3）;

由于已经已知F（0）=1，F（1）=3，F（2）=5，故我们可以这样构造

[ S（n） F(n) F(n-1) F(n-2) ]=[ S(n-1) F(n-1) F(n-2) F(n-3) ]* 4*4的构造矩阵；

1 0 0 0

故矩阵可构造为3 3 1 0

2 2 0 1

7 7 0 0；

计算结果为：S（n）=1*S(n-1)+3 * F（n-1）+2 * F（n-2）+7 * F（n-3）；(即为最后所求的结果)

F（n）=3 * F（n-1）+2 * F（n-2）+7 * F（n-3）；

F(n-1)=1*F(n-1);

F(n-2)=1*F(n-2);

代码如下：

#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
struct mat
{
    int m[4][4];
};
mat cal(mat a,mat b)
{
    mat c;
    memset(c.m,0,sizeof(c.m));//必须初始化
    for(int i=0;i<4;i++)
    for(int j=0;j<4;j++)
    for(int k=0;k<4;k++)
        c.m[i][j]=(c.m[i][j]+(a.m[i][k]*b.m[k][j])%2009)%2009;
    return c;
}
mat pow(mat a,int k)
{
    mat e;
    memset(e.m,0,sizeof(e.m));//必须初始化
    for(int i=0;i<4;i++)
        e.m[i][i]=1;
    while(k)
    {
        if(k&1) e=cal(e,a);
        a=cal(a,a);
        k>>=1;
    }
    return e;
}
int main()
{
    int T,k,ans,cas=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&k);
        if(k==0) ans=1;
        else if(k==1) ans=4;
        else if(k==2) ans=9;
        else
        {
            mat e;
            e.m[0][0]=1;e.m[0][1]=0;e.m[0][2]=0;e.m[0][3]=0;
            e.m[1][0]=3;e.m[1][1]=3;e.m[1][2]=1;e.m[1][3]=0;
            e.m[2][0]=2;e.m[2][1]=2;e.m[2][2]=0;e.m[2][3]=1;
            e.m[3][0]=7;e.m[3][1]=7;e.m[3][2]=0;e.m[3][3]=0;
            e=pow(e,k-2);
            ans=(9*e.m[0][0]+5*e.m[1][0]+3*e.m[2][0]+e.m[3][0])%2009;
        }
        printf("Case %d: %d\n",cas++,ans);
    }
    return 0;
}