一、快速幂的基本代码
int fastpow (int a, int n)
{
int ans = 1;
while(n)
{
if(n & 1) ans = ans * a % mod;
a = a * a % mod;
n = n >> 1;
}
return ans;
}
判断奇偶性用位运算效率更高:
//判断n的奇偶性
n & 1 == 1 //奇数
n & 1 == 0 //偶数
二、矩阵的快速幂
(1)重载 * (即普通的矩阵相乘)
matrix operator * (const matrix &a , const matrix &b)
{
matrix c;
memset(c.m,0,sizeof(c.m));
for(int i=0;i<N;i++)
{
for(int j=0;j<N;j++)
{
for(int k=0;k<N;k++)
{
c.m[i][j] += a.m[i][k] * b.m[k][j] % mod;//取余是为了避免数据太大
}//具体mod值参考题目要求,也有小数据的题目要求不求余
}
}
return c;
}
(2)矩阵快速幂运算(运算方法与基本代码相同,只是这里的‘*’被重载了)
matrix fastpow (matrix a ,int n)
{
matrix ans;
memset(ans.m,1,sizeof(ans.m));//初始化单位矩阵,类似普通快速幂的ans = 1;
while(n)
{
if(n & 1) ans = ans * a;//注意这里不能简写为ans *= a , 因为这里的*重载了
a = a * a;
n = n >> 1;
}
return ans;
}
三、矩阵快速幂的神奇应用(加速递推)
例:计算斐波那契数列
解法:运用矩阵改写递推关系 + 快速幂运算
注:以上内容均来自蓝桥杯罗勇军老师的网络公开课,本人对快速幂这节知识做的一点小梳理。