加强版斐波那契 数学

• 加强版斐波那契 数学
  链接：https://ac.nowcoder.com/acm/problem/15666
  来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 32768K，其他语言65536K
64bit IO Format: %lld
题目描述
这是一个加强版的斐波那契数列。
给定递推式
求F(n)的值，由于这个值可能太大，请对109+7取模。
输入描述:

第一行是一个整数T(1 ≤ T ≤ 1000)，表示样例的个数。
以后每个样例一行，是一个整数n(1 ≤ n ≤ 1018)。

输出描述:

每个样例输出一行，一个整数，表示F(n) mod 1000000007。

示例1
输入
复制

4
1
2
3
100

输出
复制

1
16
57
558616258

• 最主要的就是确定递推的矩阵
• 

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
typedef struct node 
{
	ll v[6][6];
}mat;
mat  mul(mat& A,mat& B)
{
	mat ret;
	for(int i=0;i<6;i++)
	{
		for(int j=0;j<6;j++)
		{
			ret.v[i][j]=0;
		}
	}
	for(int i=0;i<6;i++)
	{
		for(int j=0;j<6;j++)
		{
			for(int k=0;k<6;k++)
			{
				ret.v[i][j]=(ret.v[i][j]+A.v[i][k]*B.v[k][j]%mod)%mod;
			}
		}
	}
	
	return ret;
}
mat pow(mat A,ll n)
{
	mat z;
	for(int i=0;i<6;i++)
	{
	for(int j=0;j<6;j++) 
	{
		z.v[i][j]=0;
	}z.v[i][i]=1;
	}
	while(n)
	{
		if(n&1) z=mul(A,z);
		n>>=1;
		A=mul(A,A);
	}
	return z;
}
int main()
{
	int tt;
	scanf("%d",&tt);
	mat A;
	for(int i=0;i<6;i++)
	{
		for(int j=0;j<6;j++)
		{
			A.v[i][j]=0;
		}
	}
	A.v[0][0]=A.v[0][1]=A.v[0][2]=A.v[1][0]=A.v[2][2]=A.v[2][5]=A.v[3][3]=A.v[3][5]=A.v[4][4]=A.v[4][5]=A.v[5][5]=1;
	A.v[3][4]=2;
	A.v[2][3]=A.v[2][4]=3;
	A.v[0][3]=A.v[0][5]=4;
	A.v[0][4]=6;
	while(tt--)
	{
		ll n;
		scanf("%lld",&n);
		if(n==0) 
		{
			puts("0");
			continue;
		}
		if(n==1)
		{
			puts("1");
			continue;
		}
		mat ans=pow(A,n-1);
		ll res=0;
		for(int i=0;i<6;i++)
		{
			if(i==1) continue;
			res+=ans.v[0][i];
			res%=mod;
		}
		printf("%lld\n",res);
	}
	
}

simplify ----------------------------------------------------

#include<bits/stdc++.h>
using namespace std;
const int mod=1e9+7;
typedef long long ll;
struct mat 
{
	ll v[6][6];
	mat(){
		memset(v,0,sizeof(v));
	}
};
mat  mul(mat& A,mat& B)
{
	mat ret;
	for(int i=0;i<6;i++)
	{
		for(int j=0;j<6;j++)
		{
			for(int k=0;k<6;k++)
			{
				ret.v[i][j]=(ret.v[i][j]+A.v[i][k]*B.v[k][j]%mod)%mod;
			}
		}
	}
	
	return ret;
}
mat pow(mat A,ll n)
{
	mat z;
	for(int i=0;i<6;i++)
	{
      z.v[i][i]=1;
	}
	while(n)
	{
		if(n&1) z=mul(A,z);
		n>>=1;
		A=mul(A,A);
	}
	return z;
}
int main()
{
	int tt;
	scanf("%d",&tt);
	mat A;
	A.v[0][0]=A.v[0][1]=A.v[0][2]=A.v[1][0]=A.v[2][2]=A.v[2][5]=A.v[3][3]=A.v[3][5]=A.v[4][4]=A.v[4][5]=A.v[5][5]=1;
	A.v[3][4]=2;
	A.v[2][3]=A.v[2][4]=3;
	A.v[0][3]=A.v[0][5]=4;
	A.v[0][4]=6;
	while(tt--)
	{
		ll n;
		scanf("%lld",&n);
		if(n==0) 
		{
			puts("0");
			continue;
		}
		if(n==1)
		{
			puts("1");
			continue;
		}
		mat ans=pow(A,n-1);
		ll res=0;
		for(int i=0;i<6;i++)
		{
			if(i==1) continue;
			res+=ans.v[0][i];
			res%=mod;
		}
		printf("%lld\n",res);
	}
	
}