Count “字节跳动-文远知行杯”广东工业大学第十四届程序设计竞赛

Problem Description

Farmer John有n头奶牛.
某天奶牛想要数一数有多少头奶牛,以一种特殊的方式:
第一头奶牛为1号,第二头奶牛为2号,第三头奶牛之后,假如当前奶牛是第n头,那么他的编号就是2倍的第n-2头奶牛的编号加上第n-1头奶牛的编号再加上自己当前的n的三次方为自己的编号.
现在Farmer John想知道,第n头奶牛的编号是多少,估计答案会很大,你只要输出答案对于123456789取模.

Input

第一行输入一个T,表示有T组样例
接下来T行,每行有一个正整数n,表示有n头奶牛 (n>=3)
其中，T=10^4,n<=10^18

Output

共T行,每行一个正整数表示所求的答案

Sample Input
5
3
6
9
12
15
 

Sample Output
31
700
7486
64651
527023

思路：这道题的递推式 F(n)=2*F(n-2)+F(n-1)+n*n*n;

一看这就是矩阵快速幂 

首先我们不知道递推矩阵是几乘几的

我们只知道

                    (n+1)^3=n^3 +  3*n^2 +3*n + 1;

 所以  f.a 应该是

  

这样就可以推出递推矩阵 base 为 6* 6

所以下面直接就可以用矩阵快速幂了

代码如下：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int mod=123456789;
struct Marx
{
    ll a[6][6];
}base,f,imat;
Marx mul(Marx x,Marx y)
{
    Marx ret;
    memset(ret.a,0,sizeof(ret.a));
    for(int i=0;i<6;i++)
    {
        for(int k=0;k<6;k++)
        {
            for(int j=0;j<6;j++)
            {
                ret.a[i][j]=(ret.a[i][j]+(x.a[i][k]*y.a[k][j])%mod)%mod;
            }
        }
    }
    return ret;
}
Marx ksm(ll x)
{
    if(!x)return imat;
    Marx tmp=ksm(x>>1);
    if(x&1)return mul(mul(tmp,tmp),base);
    return mul(tmp,tmp);

}
int main()
{
     memset(base.a,0,sizeof(base.a));
     base.a[0][1]=base.a[1][1]=base.a[1][2]=base.a[2][2]=base.a[2][5]=base.a[3][3]=base.a[3][5]=base.a[4][4]=base.a[4][5]=base.a[5][5]=1;
     base.a[1][0]=base.a[3][4]=2;
     base.a[2][3]=base.a[2][4]=3;
     memset(f.a,0,sizeof(f.a));
     f.a[0][0]=1;f.a[1][0]=2;f.a[2][0]=27;f.a[3][0]=9;f.a[4][0]=3;f.a[5][0]=1;
     for(int i=0;i<6;i++)imat.a[i][i]=1;
     int t;
     scanf("%d",&t);
     while(t--)
     {
         ll n;
         scanf("%lld",&n);
         Marx ans;
         memset(ans.a,0,sizeof(ans));
         ans=mul(ksm(n-2),f);
         printf("%lld\n",ans.a[1][0]);
     }
     return 0;
}