Hrbust 1401 九连环【递推+矩阵快速幂】

九连环

Time Limit: 1000 MS Memory Limit: 65536 K Total Submit: 66(38 users) Total Accepted: 35(32 users) Rating: Special Judge: No

Description

相信大家都玩过九连环的游戏 ，九连环的规则是 ： 1第一个环可以在任何时候挂到柄上或者从柄上摘下 2在任何时候，你只能操作一个环 3如果前k-2个环都没有在柄上，并且第k-1个环在柄上，这时如果第k个环在柄上的话，可以把它摘下来，如果它不在柄上，可以把它挂上去

Input

给定一个数n，n不大于10的8次方，你的任务是输出从柄上摘下n个环需要的最小操作次数，测试数据不超过100组。

Output

输出1行，因为结果非常大，所以你需要输出结果对素数10007的余数。

Sample Input

1 2 9

Sample Output

1 2 341

思路：

暴力打表找递推式、

F【i】=F【i-1】+F【i-2】*2+1；

n比较大，矩阵快速幂加速优化即可。

Ac代码：

//F[n]=F[n-1]+F[n-2]*2+1;
#include<stdio.h>
#include<string.h>
using namespace std;
#define ll long long int
int const MOD = 10007;
struct matrix
{
    ll m[3][3];
};
matrix A,B,tmp;
matrix multiply(matrix x, matrix y)  //矩阵乘法
{
    matrix tmp;
    memset(tmp.m, 0, sizeof(tmp.m));
    for(int i = 0; i < 3; i++)
    {
        for(int j = 0; j < 3; j++)
        {
            if(x.m[i][j] == 0)
                continue;
            for(int k = 0; k < 3; k++)
            {
                if(y.m[j][k] == 0)
                    continue;
                tmp.m[i][k] += x.m[i][j] * y.m[j][k] % MOD;
                tmp.m[i][k] %= MOD;
            }
        }
    }
    return tmp;
}

matrix quickmod(matrix a, int n) //矩阵快速幂
{
    matrix res;
    memset(res.m, 0, sizeof(res.m));
    for(int i = 0; i < 3; i++)
        res.m[i][i] = 1;
    while(n)
    {
        if(n & 1)
            res = multiply(res, a);
        n >>= 1;
        a = multiply(a, a);
    }
    return res;
}
int main()
{
    ll n;
    while(~scanf("%lld",&n))
    {
        if(n==1)
        {
            printf("1\n");
            continue;
        }
        if(n==2)
        {
            printf("2\n");
            continue;

        }
        memset(A.m,0,sizeof(A.m));
        A.m[0][0]=1;A.m[0][1]=2;A.m[0][2]=1;
        A.m[1][0]=1;A.m[1][1]=0;A.m[1][2]=0;
        A.m[2][0]=0;A.m[2][1]=0;A.m[2][2]=1;
        memset(tmp.m,0,sizeof(tmp.m));
        tmp.m[0][0]=2;
        tmp.m[1][0]=1;
        tmp.m[2][0]=1;
        B=quickmod(A,n-1);
        B=multiply(tmp,B);
        ll ans=B.m[0][0];
        if(n%2==1)ans=(ans-1);
        ans=(ans%MOD+MOD)%MOD;
        printf("%lld\n",ans);
    }
}
/*
f[n-1]      1   2   1   F[n]
F[n-2]      1   0   0   F[n-1]
1           0   0   1   1
*/
/*
1
2
5
10
21
42
85
170
*/