ACM_快速幂

来源:互联网 发布:程序员述职报告范文 编辑:程序博客网 时间:2024/06/05 04:52

快速计算一个数的多少次幂(底数一半是整数, 如果是浮点数, 建议直接用库函数的pow, 指数必须是整数)

ll exp_mod(ll a, ll b){    ll ret = 1;    while(b){        if(b & 1) ret = ret * a;        a = a * a;        b >>= 1;    }    return ret;}

程序解释:
比如3 ^ 5 = 243
快速幂是指数基于二进制的计算 所以把他写成这样 3 ^ 101 (101是二进制)
那么快速幂的计算过程是这样的:
1. 结果用ret记录(初始为1)
2.我们发现指数的二进制下的最后1位是1说明这一位有这个数, 然后ret = ret * a , ret = 3;
3.然后我们把指数右移一位, 指数(也就是程序中的b) = 10(二进制) 同时对应的k 也应该”右移一位”指向101中的0 如何操作呢? 只需要a = a * a, a = 9;
返回步骤2
再次循环, 直到指数为0(因为指数一直右移, 所以肯定能为0);
所以3 ^ 5 的实际过程

初始化 k = 3, b = 101, a = 3, ret = 1;
第一次循环{
发现b的最低位是1 ret = ret * a , ret = 3;
a = a * a, a = 9;
b = b >> 1, b = 10(二进制)
}
第二次循环{
发现b的最低位是0 ret不变
a = a * a, a = 81;
b = b >> 1, b = 1;
}
第三次循环{
发现b的最低位是1, ret = ret * a, ret = 243;
a = a * a, a = 243;
b = b >> 1, b = 0;
发现b等于0 所以在这次循环结束后退出
}

返回ret也就是243, 于是得到3 ^ 5 = 243;
整个算法的效率是log N, 应该比库函数的pow快吧, 而且最主要的没有精度损失!!!

可以加上mod(需要取余的数)

ll exp_mod(ll a, ll b, ll mod){    ll ret = 1;    a %= mod;    while(b){        if(b & 1) ret = (ret * a) % mod;        a = a * a % mod;        b >>= 1;    }    return ret;}
0 0
原创粉丝点击