A Short problem(矩阵快速幂+循环节)

题目链接

感谢博主HDU 4291 A Short problem(找循环节 快速幂)

题意

给你n求g(g(g(n)))mod109+7g(g(g(n)))mod109+7的值，其中g(x)=3g(x−1)+g(x−2)g(x)=3g(x−1)+g(x−2)
思路

这个问题应该从最外层向里思考，对于$$g(x)=3g(x-1)+g(x-2)g(x)=3g(x-1)+g(x-2)$$来说对$$1e9+7$$取余的话，根据取模和递推方程的特性当再次出现连着的两项是0和1的话那么接下去的项数就会之前的重复了,那我们可以这样考虑因为g(x)出现了循环节那么g(x%循环节)=g(x)g(x%循环节)=g(x)由于n的范围是1到1e18所以我们求循环节可以这样求

矩阵
|g[n-1] g[n-2]|*|3 1|=|g[n] g[n-1]|
                |1 0|
gg(x)=g(g(x));
由于用g(x)求gg(x),g(x)不能对1e9+7取余，导致g(x)很大，所以不能直接求
输出结果可发现g(x),gg(x)都是循环的,暴力求出循环。
在%1e9+7的基础上g(g(g(x)))=g(g(g(x))+222222224),
在%222222224的基础上g(g(x))=g(g(x)+183120);
用mod={183120，222222224，1e9+7}循环三次矩阵乘法即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<vector>
using namespace std;

typedef long long ll;
ll n;
int mod[3] = {1e9+7,222222224,183120};
ll ans[2][2],tem[2][2];//可以在下面矩阵快速幂的函数中对比快速幂看

void jzxc(ll a[2][2],ll b[2][2],int modd)//矩阵相乘
{
    ll c[2][2];
    memset(c,0,sizeof(c));
    for(int i = 0; i < 2; i++)
    {
        for(int j = 0; j < 2; j++)
        {
            for(int k = 0; k < 2; k++)
            {
                c[i][j] += a[i][k]*b[k][j];
            }
        }
    }
    for(int i=0; i<2; i++)
    {
        for(int j=0; j<2; j++)
        a[i][j]=c[i][j]%modd;
    }
}

ll jzksm(ll k,int modd)//矩阵快速幂
{
    tem[0][0] = 3;
    tem[1][1]=0;
    tem[0][1]=tem[1][0]=1;
    memset(ans,0,sizeof(ans));
    ans[0][0] = ans[1][1] = 1;
    while(k)
    {
        if(k&1) jzxc(ans,tem,modd);
        jzxc(tem,tem,modd);
        k >>= 1;
    }
    return ans[0][0];
}

int main()
{
    while(cin >> n)
    {
        if(n >= 2) n = jzksm(n-1,mod[2]);//因为0,1的值我们都知道了所以判断条件
        if(n >= 2) n = jzksm(n-1,mod[1]);//为是否n >= 2
        if(n >= 2) n = jzksm(n-1,mod[0]);
        printf("%lld\n",n);
    }
    return 0;
}
//#include<bits/stdc++.h>
//using namespace std;
//
//typedef long long ll;
//
//ll mod1 = 1e9+7;//g(g(g(n))每mod1一循环
//ll mod2 = 222222224;//g(g(n))每mod2一循环
//ll mod3 = 183120;//g(n)每mod3一循环
//ll mod4 = 240;//n每mod4一循环
//ll mmax = 1e18;
//int main()
//{
//    ll a = 1;
//    ll b = 0;
//    for(ll i = 1; i <= mmax; i++)
//    {
//        ll temp = (3*a%mod3+b%mod3)%mod3;
//        b = a;
//        a = temp;
//        if(a == 1 && b == 0)
//        {
//            printf("%lld\n",i);
//            break;
//        }
//    }
//    return 0;
//}