题目链接:https://ac.nowcoder.com/acm/contest/6631/C
- 直接矩阵快速幂就可。
- 基础矩阵,
- 那么
AC代码:
typedef long long LL;
const int mod = 1e9+7;
class Solution {
public:
struct mat{
LL m[2][2];
};
mat muti(mat a,mat b)//矩阵乘法
{
mat tmp;
for(int i=0;i<2;i++)
{
for(int j=0;j<2;j++)
{
tmp.m[i][j]=0;
for(int k=0;k<2;k++)
{
tmp.m[i][j]+=a.m[i][k]*b.m[k][j];
tmp.m[i][j]%=mod;
}
}
}
return tmp;
}
LL mat_pow(mat a,LL n)//矩阵快速幂
{
mat ans={1,0,0,1};
while(n)
{
if(n&1)ans=muti(ans,a);
a=muti(a,a);
n>>=1;
}
return ans.m[0][0];
}
long long nthElement(long long n, long long b, long long c) {
if(n==0)return 0;
if(n==1)return 1;
mat base={b,c,1,0};
return mat_pow(base, n-1);
}
};