acm数学（1）快速幂

经常会遇到对一个数进行幂运算，如求$x^n$，朴素的算法是把这个数乘n次
，复杂度为O(n)，在n比较大的情况就会非常慢。而快速幂可以在O(logn)的复杂度内计算乘方。
快速幂用到的是二进制的思想。比如要算出$5^{31}$，我们把31转换为2进制，可以转化为$5^{(11111{)}_2}=5^{2^4}\cdot 5^{2^3}\cdot 5^{2^2}\cdot 5^{2^1}\cdot 5^{2^0}$，而$x^{2^{i+1}}=(x^{2^i}{)}^2$，于是我们只需要不断把底数取平方再把各项乘起来就好了。

代码如下

int qpow(int x, int n){
    int ans = 1;
    while(n){
        if(n & 1) ans *= x;//位运算，判断n的末位是否为1
        //如果满足则让ans乘当前的x底数
        x *= x ;//底数取平方
        n >>= 1;//让n右移一位比如1010右移变成了0101相当于整除以二
    }
    return ans;
}

我们可以根据代码推敲一下这个过程，例如$5^{29}$
$5^{(29{)}_{10}}=5^{(11101{)}_2}$
传入函数后
n = 11101, x = 5,
① n & 1 = 1, ans = ans * x = 1 * 5 = 5, x = x * x = $5^2$ , n = n>>1 = 1110
②n & 1 = 0, x = x * x = $5^4$, n = n>>1 = 111
③n & 1 = 1, ans = ans * x = $5\ast 5^4=5^{(101{)}_2}$, x = x * x = $5^8$, n = n>>1 = 11;
④n & 1 = 1, ans = ans * x = $5^{(101{)}_2}\ast 5^8=5^{(1101{)}_2}$, x = x * x = $5^{16}$, n = n>>1 = 1;
⑤n & 1 = 1, ans = ans * x = $5^{(1101{)}_2}\ast 5^{16}=5^{(11101{)}_2}=5^{29}$, x = x * x = $5^{32}$, n = n>>1 = 0
n = 0，跳出while循环，返回答案ans = $5^{29}$。

由于n经常很大，涉及取模，那么我们的模板可以改为

long long qpow(long long x, long long n){
    long long ans = 1;
    while(n){
        if(n & 1) ans = (x * ans) % p;
        x = (x * x) % p;
        n >>= 1;
    }
    return ans % p;
}

其中p为模

此外，快速幂也可以用在对矩阵进行幂运算上。只要底数满足乘法和乘法结合律，快速幂应该都是有效的。