題目連結

題意：給一堆公式，求⌊a[n]‾‾‾‾√⌋的值。

官方題解的公式推的實在是沒什麼道理，但是這個xjb亂猜我真的是一口老血噴出來。。。

令f(n)=⌊a[n]‾‾‾‾√⌋，則f(n)=4f(n−1)+17f(n−2)−12f(n−3)

沒錯，就是h(n)的那個遞推……

然後就用矩陣快速冪瞎搞搞就好了，沒什麼可說的。

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

typedef long long ll;
const ll mod=1000000007;

struct Matrix{
    ll a[5][5];
    void init(){
        for
 
 (int i=1;i<=3;i++){
            for (int j=1;j<=3;j++){
                a[i][j]=0;
            }
        }
    }
    Matrix operator * (const Matrix &b) const{
        Matrix tmp;
        tmp.init();
        for (int i=1;i<=3;i++){
            for (int j=1;j<=3;j++){
                for
 
 (int k=1;k<=3;k++){
                    tmp.a[i][j]+=(a[i][k]*b.a[k][j]%mod+mod)%mod;
                }
            }
        }
        return tmp;
    }
};

Matrix mat_pow(Matrix mat,ll n){
    Matrix ans;
    ans.init();
    ans.a[1][1]=ans.a[2][2]=ans.a[3][3]=1;
    while (n){
        if (n&1
 
)    ans=mat*ans;
        mat=mat*mat;
        n>>=1;
    }
    return ans;
}

int main(){
    int T;
    scanf("%d",&T);
    while (T--){
        ll n;
        scanf("%lld",&n);
        if (n==2)   printf("31\n");
        else if (n==3)  printf("197\n");
        else if (n==4)  printf("1255\n");
        else{
            Matrix mat;
            mat.init();
            mat.a[1][1]=4,mat.a[1][2]=17,mat.a[1][3]=-12;
            mat.a[2][1]=1,mat.a[3][2]=1;
            mat=mat_pow(mat,n-4);
            ll ans=((mat.a[1][1]*1255%mod+mat.a[1][2]*197%mod+mod)%mod+31*mat.a[1][3]%mod+mod)%mod;
            ans%=mod;
            printf("%lld\n",ans);
        }
    }
}