粉樱花之恋(矩阵快速幂求斐波拉契数列)

qn是个特别可爱的小哥哥,qy是个特别好的小姐姐,他们两个是一对好朋友 [ cp (划掉~)
又是一年嘤花烂漫时,小qn于是就邀请了qy去嘤花盛开的地方去玩。当qy和qn来到了田野里时,qy惊奇的发现,嘤花花瓣以肉眼可见的速度从树上长了出来。
仔细看看的话,花瓣实际上是以一定规律长出来的,而且,每次张成新的花瓣的时候,上一次的花瓣就会都落到地上,而且不会消失。
花瓣生长的规律是,当次数大于等于2时,第i次长出来的花瓣个数和上一次张出来的花瓣个数的差是斐波那契数列的第i-1项。初始的时候地上没有花瓣,树上的花瓣个数为1,第一次生长的花瓣个数为1。初始的那个花瓣就落到了地上

现在,小qn想知道,经过k次生长之后,树上和地上的总花瓣个数是多少?

ps:斐波那契数列:

    f[1]=f[2]=1;f[i]=f[i-1]+f[i-2] (i>=2且i   N+) 

输入描述:

一行一个数k

输出描述:

一行一个数m,表示第k次生长过后,树上和地上的总花瓣数是多少。由于答案会很大,请你将答案mod 998244353后输出

示例1

输入

4

输出

12

说明

第一次:树上1,地上1.第二次树上2,地上1+1,第三次树上3,地上1+1+2,第四次树上5,地上1+1+2+3。总共12个

示例2

输入

5

输出

20

说明

第五次树上8,地上1+1+2+3+5。总共20个

备注:

对于0%的数据,有k=样例
对于20%的数据,有k<=1'000
对于60%的数据,有k<=1'000'000
对于80%的数据,有k<=1'000'000'000
对于100%的数据,有k<1'000'000'000'000'000'000

思路:关于斐波拉契数列的一些性质:https://www.cnblogs.com/Milkor/p/4734763.html,那么前n项之和即为第n+2项-1的值。

所以这个题就十分明朗了。虽然简单,,但是一直不会构造初始矩阵,这次终于算是搞懂了。(线代哟~~~)

粉樱花之恋(矩阵快速幂求斐波拉契数列)_第1张图片

粉樱花之恋(矩阵快速幂求斐波拉契数列)_第2张图片

就这样了......

该题代码如下:

#include
#include
#include
#define mod 998244353 
struct node
{
    long long v[3][3];
    long long int m,l;
};
node get_mul(node a,node b)
{
    node c;
    c.m=a.m;c.l=b.l;
    for(long long int i=1;i<=c.m;i++)
      for(long long int j=1;j<=c.l;j++)
      {
          c.v[i][j]=0;
          for(long long int k=1;k<=a.l;k++)
              c.v[i][j]=(c.v[i][j]+a.v[i][k]*b.v[k][j])%mod;
      }
    return c;
}

int main()
{
        long long int n;
        scanf("%lld",&n);
        if(n==0) {printf("1\n");return 0;}//注意这个特判0
        node a,b,c;
        a.m=a.l=2,a.v[1][1]=1,a.v[1][2]=1,a.v[2][1]=1,a.v[2][2]=0;
        b.m=b.l=2,b.v[1][1]=1,b.v[1][2]=0,b.v[2][1]=0,b.v[2][2]=1;
        c.m=2,c.l=1,c.v[1][1]=1,c.v[2][1]=0;
        n+=2;
        while(n)
        {
            if(n&1) b=get_mul(a,b);
            a=get_mul(a,a);
            n>>=1;
        }
        b=get_mul(b,c);
        printf("%lld\n",b.v[1][1]-1);
    return 0;
}

 

你可能感兴趣的:(快速幂/快乘,矩阵快速幂,斐波拉契数列)