牛客算法周周练19-C、粉嘤花之恋【斐波那契数列求和、矩阵快速幂】

题目链接
题意：
花瓣以一定规律长出，而且，每次长成新的花瓣的时候，上一次的花瓣就会都落到地上，而且不会消失。
花瓣生长的规律是，当次数大于等于2时，第i次长出来的花瓣个数和上一次长出来的花瓣个数的差是斐波那契数列的第i-1项。初始的时候地上没有花瓣，树上的花瓣个数为1，第一次生长的花瓣个数为1。初始的那个花瓣就落到了地上
现在，小qn想知道，经过k次生长之后，树上和地上的总花瓣个数是多少?
其实就是求斐波那契数列前k+1项和
思路：需要了解斐波那契求和公式：$$\sum _{i=1}^{n}F(i)=F(i+2)-1$$
然后我们需要了解如何快速求斐波那契数列的第n项，这里使用矩阵快速幂即可，不太明白的可以搜索一下，有很多讲得很好的教程，由于时间有限，暂时先留着。
AC代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;

const int N = 6e6 + 10;
const int MAX = 500;
const ll mod1 = 1e9 + 7;
const ll mod2 = 1e9 + 9;
const ull P1 = 233;
const ll P2 = 1331;
const int mod = 998244353;
const double PI = acos(-1.0);
const ll INF = 0x3f3f3f3f3f3f3f;
int T;
struct matrix{
    ll m[2][2];
}ans, base;

matrix multi(matrix a, matrix b){
    matrix tmp;
    for(int i = 0; i < 2; ++i)
    {
        for(int j = 0; j < 2; ++j)
        {
            tmp.m[i][j] = 0;
            for(int k = 0; k < 2; ++k)
                tmp.m[i][j] = (tmp.m[i][j] + a.m[i][k] * b.m[k][j]) % mod;
        }
    }
    return tmp;
}
ll fast_mod(ll n)
{
    base.m[0][0] = base.m[0][1] = base.m[1][0] = 1;
    base.m[1][1] = 0;
    ans.m[0][0] = ans.m[1][1] = 1;
    ans.m[0][1] = ans.m[1][0] = 0;
    while(n){
        if(n & 1)
            ans = multi(ans, base);
        base = multi(base, base);
        n >>= 1;
    }
    return ans.m[0][1];
}

signed main()
{
    ll n;
    cin>>n;
    printf("%d\n", (fast_mod(n+3)-1+mod)%mod);
    return 0;
}