用快速幂求矩阵乘法
代码如下:
struct matrix
{
int m[2][2];
};
class Solution {
public:
matrix multi(matrix a, matrix b)
{
matrix ans;
for(int i=0; i<2; i++)
{
for(int j=0; j<2; j++)
{
ans.m[i][j]=0;
for(int k=0; k<2; k++)
{
ans.m[i][j]+=a.m[i][k]*b.m[k][j];
}
}
}
return ans;
}
int fibonacci(int n) {
n--;
matrix base,ans;
base.m[0][0]=base.m[0][1]=base.m[1][0]=1;
base.m[1][1]=0;
ans.m[0][0]=ans.m[1][1]=1;
ans.m[1][0]=ans.m[0][1]=0;
while(n)
{
if(n&1) ans=multi(ans,base);
base=multi(base,base);
n>>=1;
}
return ans.m[0][1];
}
};