简单计数（矩阵快速幂）

题源来自牛客https://ac.nowcoder.com/acm/problem/24639
题意;（题意未改动，题面已改动）
有一个长度为k的队列满足以下条件：
①第一个元素不能为1，,最末一个元素必须为1；
②任意两个相邻元素都不相同
③这个数列中的所有数必须小于等于n
问：满足条件的组合数
题解：
设F[0][i]表示从第一个元素到第i个元素，前i-1个元素满足条件，而第i个元素非1的组合数
F[1][i]反之表示第i个元素为i的组合数

那么我们要求的答案就是F[1][k]
那么我们可以得到转移方程：
F[0][i]=(n-2)F[0][i-1]+(n-1)F[1][i-1]
F[1][i]=F[0][i-1]
从而F[0][i]=(n-1)F[0][i-1]+(n-1)F[0][i-2];
F[1][k]=F[0][k-1];
？？？？所以说与第一维完全无关！！这不就是一个炒鸡简单的矩阵快速幂吗？？

附上源码

#include<bits/stdc++.h>
using namespace std;
#define lowerbit(x) ((-x)&(x))
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define MS(x,y) memst(x,y,sizeof(x))
typedef long long ll;
const int maxn=3;
const int mod=998244353;
struct Matrix{
    ll f[maxn][maxn],n,m;
    Matrix(){memset(f,0,sizeof f);}
    void O(){memset(f,0,sizeof f);}
    void E(){fo(i,1,m)f[i][i]=1;}
};
inline Matrix operator *(Matrix a,Matrix b){
    Matrix c;c.n=a.n;c.m=b.m;
    fo(i,1,c.n)fo(j,1,c.m)fo(k,1,b.n)
    c.f[i][j]=(c.f[i][j]+a.f[i][k]*b.f[k][j]%mod)%mod;
    return c;
}
inline Matrix power(Matrix a,ll p){
    Matrix c;c.n=a.n;c.m=a.m;c.E();
    while(p){
        if(p&1)c=c*a;
        a=a*a;
        p>>=1;
    }
    return c;
}
int main(){
    ll n,k;
    while(~scanf("%lld%lld",&n,&k)){
    if(k==0||k==1){
        cout<<0<<endl;
        continue;
    }
    if(k==2){
        cout<<(n-1)%mod<<endl;
        continue;
    }
    Matrix M;
    M.n=M.m=2;
    M.f[1][1]=n-2;M.f[1][2]=n-1;M.f[2][1]=1;
    M=power(M,k-2);
    cout<<M.f[1][1]*(n-1)%mod<<endl;
    }
    return 0;
}