快速幂 python代码求输入两个整数a、b,求a b 。结果保证在long long int范围内。
时间: 2023-06-07 13:12:02 浏览: 167
Python 整数数据类型详解(int)[学习 Python 必备基础知识][看此一篇就够了]
好的,我知道了。下面是快速幂的 Python 代码:
```python
def pow(a, b, p):
res = 1
while b > 0:
if b & 1:
res = res * a % p
a = a * a % p
b >>= 1
return res
a, b = map(int, input().split())
p = 10 ** 9 + 7
print(pow(a, b, p))
```
使用时,输入两个整数 a 和 b,程序会快速求出 a 的 b 次方的余数(对 $10^9+7$ 取模)。如果需要求出非取模的结果,可以将函数返回语句中的 `% p` 删去。
阅读全文