题目:点击打开链接
每次福州的最担心的就是暴0,这次居然又是暴0,说到底还是水平太低了,有思路的总是wa,tle,代码不够完善
最后的公式是 a[1]=1,a[2]=7,a[n]=6*a[n-1]-a[n-2]+1,显然是用矩阵快速幂
提交的时候记得用Visual C++,不要用GNU C++,否则会超时
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
using namespace std;
#define mod 1000000007
#define ll __int64
template <class T>
inline bool rd(T &ret)
{
char c;
int sgn;
if(c=getchar(),c==EOF) return 0;
while(c!='-'&&(c<'0'||c>'9')) c=getchar();
sgn=(c=='-')?-1:1;
ret=(c=='-')?0:(c-'0');
while(c=getchar(),c>='0'&&c<='9') ret=ret*10+(c-'0');
ret*=sgn;
return 1;
}
template <class T>
inline void pt(T x)
{
if (x <0)
{
putchar('-');
x = -x;
}
if(x>9) pt(x/10);
putchar(x%10+'0');
}
struct matrix
{
int a[3][3];
};
matrix mul(matrix a,matrix b)
{
matrix ret;
memset(ret.a,0,sizeof(ret.a));
for(int i=0;i<3;i++)
{
for(int j=0;j<3;j++)
{
if(a.a[i][j])
{
for(int k=0;k<3;k++)
{
ret.a[i][k]= (ret.a[i][k]+(ll)a.a[i][j]*b.a[j][k])%mod;
}
}
}
}
return ret;
}
ll B[3]={7,1,1};
int main()
{
int t;
matrix tmp;
tmp.a[0][0]=6,tmp.a[0][1]=-1,tmp.a[0][2]=1;
tmp.a[1][0]=1,tmp.a[1][1]=0,tmp.a[1][2]=0;
tmp.a[2][0]=0,tmp.a[2][1]=0,tmp.a[2][2]=1;
matrix h[80];
memset(h[0].a,0,sizeof(h[0].a));
h[0].a[0][0]=h[0].a[1][1]=h[0].a[2][2]= 1;
h[1]=tmp;
for(int i=2;i<65;i++) h[i]=mul(h[i-1],h[i-1]);
rd(t);
while(t--)
{
ll n;
rd(n);
if(n==1)
{
pt(6);
putchar('\n');
continue;
}
n--;
matrix e=h[0];
for(int i=1;n;n>>=1,i++)
{
if(n&1) e=mul(e,h[i]);
}
ll ans=0;
for(int i=0;i<3;i++)
{
ans=(ans+e.a[0][i]*B[i])%mod;
}
ans--;
ans=(ans+mod)%mod;
pt(ans);
putchar('\n');
}
return 0;
}