python 快速幂算法

计算a的b次方,需要考虑:

1、b==0 的话,返回1

2、b<0的情况,设置一个flag,用于返回数据的时候进行判断

3、b>0的情况:

打个比方:2^6=2*2*2*2*2*2=(2*2)*(2*2)*(2*2)=(2*2)^3=(2*2)*(2*2)^2=

# -*- coding:utf-8 -*-
class Solution:
    def Power(self, base, exponent):
        # write code here
        flag=1
        re=1
        tmp=base
        if exponent==0:   #等于0的情况
            return 1
        if exponent<0:   #小于0的时候,设置一个flag,用于返回的时候做判断
            flag=0
            exponent=abs(exponent)

        while exponent>0:    #大于0的情况
            if exponent&1==1:
                re=re*tmp     #如果是奇数,奇数减一就是偶数
            exponent>>=1      #右移动一位就是除以2
            tmp=tmp*tmp       #2^6=(2*2)^3
        return re if flag else 1/re

猜你喜欢

转载自blog.csdn.net/xiaolangmin/article/details/89487277