hdu6470 Count(矩阵快速幂)

题目

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

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

题解

心得

虽然是自己A的但还是想再总结一下啊

顺便把矩阵快速幂的板子最终确定下来

以后敲就用这个板子了

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=7;
const int mod=123456789;
struct mat
{
    ll a[maxn][maxn];
    mat()
    {
        memset(a,0,sizeof a);
    }
    void clear()
    {
        memset(a,0,sizeof a);
    }
};
mat operator*(mat a,mat b)
{
    mat c;
    for(int i=0;i<=5;++i)
    {
        for(int j=0;j<=5;++j)
        {
            for(int k=0;k<=5;++k)
            {
                c.a[i][j]+=a.a[i][k]%mod*b.a[k][j]%mod;
                if(c.a[i][j]>=mod)c.a[i][j]%=mod;
            }
        }
    }
    return c;
}
mat modpow(mat x,ll p)
{
    mat res;
    for(int i=0;i<=5;++i)
    res.a[i][i]=1;
    for(;p;x=x*x,p/=2)
    if(p&1)res=res*x; 
    return res;
} 
mat b;
ll n,a[maxn];
int T;
void init()
{
    b.clear();
    b.a[0][0]=1;b.a[0][1]=2;b.a[0][2]=1;
    b.a[1][0]=1;
    b.a[2][2]=b.a[2][5]=1;
    b.a[2][3]=b.a[2][4]=3;
    b.a[3][3]=b.a[3][5]=1;b.a[3][4]=2;
    b.a[4][4]=b.a[4][5]=1;b.a[5][5]=1;
    a[0]=2,a[1]=1,a[2]=27,a[3]=9,a[4]=3,a[5]=1;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        init();
        scanf("%lld",&n);
        /*if(n==1||n==2)
        {
            printf("%lld\n",n);
            continue;
        }*/
        mat ans;
        for(int i=0;i<=5;++i)
        ans.a[i][i]=1;
        ans=modpow(b,n-2);
        ll res=0;
        for(int i=0;i<=5;++i)
        {
         res+=ans.a[0][i]%mod*a[i]%mod;
         if(res>=mod)res%=mod;
        }
        printf("%lld\n",res);
    }
    return 0;
}