hrbust 1430 哈理工oj 1430 神秘植物【递推+矩阵快速幂】

神秘植物

Time Limit: 1000 MS Memory Limit: 65536 K Total Submit: 55(26 users) Total Accepted: 24(20 users) Rating:  Special Judge: No

Description

Leyni有一种神秘的植物，形状是一种“向上”三角形植物，每过一年，每个“向上”三角形植物会变成三个“向上”三角形植物和一个“向下”三角形植物，每个“向下”三角形植物会变成三个“向下”三角形植物和一个“向上”三角形植物。如下图： Leyni想知道经过n年有多少个“向上”三角形植物。

Input

输入包含多组测试数据。 对于每组测试数据： 第1行，包含一个整数n (0 ≤ n ≤ 1018) 处理到文件结束

Output

对于每组测试数据： 第1行，输出Leyni拥有多少个“向上”三角形植物。(MOD 1000000007)

Sample Input

1 2

Sample Output

3 10

我们通过枚举样例的方式来寻找规律：

我们用Fn表示蓝色（向上）的三角形，用Sn来表示白色（向下）的三角形，慢慢寻求规律，我们能推出来蓝色和白色小三角形之间的依赖关系如下

N	0	1	2	3	4
Fn	1	3	10	36	36*3+28
Sn	0	1	6	28	28*3+36

我在写F4和S4的时候已经把规律写出来了即：

Fn=Fn-1*3+Sn-1

Sn=Sn-1*3+Fn-1

找到规律之后看N有多大，如果N是int范围之内的我们还可以考虑考虑用O（n）来解题，但是这个题N有10^18大 ，即使是O（n）也一样会超时，所以我们采用矩阵快速幂的形式来求解。

写出式子如下求出矩阵：

（	?	?	）	X	(	Fn-1	)	=	(	Fn	)
(	?	?	)		(	Sn-1	)	=	(	Sn	)

根据矩阵乘法公式，简单推出矩阵结果：

3 1

1 3

AC代码：

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
#define mod 1000000007
#define ll long long int
typedef struct Matrix
{
    ll mat[2][2];
}matrix;
matrix A,B;
Matrix matrix_mul(matrix a,matrix b)
{
    matrix c;
    memset(c.mat,0,sizeof(c.mat));
    int i,j,k;
    for(int i=0;i<2;i++)
    {
        for(int j=0;j<2;j++)
        {
            for(int k=0;k<2;k++)
            {
                c.mat[i][j]+=a.mat[i][k]*b.mat[k][j];
                c.mat[i][j]%=mod;
            }
        }
    }
    return c;
}
Matrix matrix_quick_power(matrix a,ll k)//矩阵快速幂0.0
{
    matrix b;
    memset(b.mat,0,sizeof(b.mat));
    for(int i=0;i<2;i++)
    b.mat[i][i]=1;//单位矩阵b
    while(k)
    {
        if(k%2==1)
        {
            b=matrix_mul(a,b);
            k-=1;
        }
        else
        {
            a=matrix_mul(a,a);
            k/=2;
        }
    }
    return b;
}
int main()
{
    ll n;
    while(cin>>n)
    {
        A.mat[0][0]=3;A.mat[0][1]=1;
        A.mat[1][0]=1;A.mat[1][1]=3;
        B=matrix_quick_power(A,n);
        cout<<B.mat[0][0]<<endl;
    }
}