STL power() 计算一个数的n次幂

最新推荐文章于 2024-09-28 11:37:12 发布

原创最新推荐文章于 2024-09-28 11:37:12 发布 · 1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#stl #power

STL源码学习专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种使用非递归二分法实现的STL幂运算函数，该方法通过将指数转换为二进制并从低位到高位依次相乘来高效计算x的n次幂，时间复杂度为O(logn)。

计算一个数的n次幂

STL计算x的n次幂的函数实现原理：用非递归的二分法
比如

x^29 = x^1 * x^4 * x^8 * x^16;
29 = (11101)B

也就是把指数转换为二进制的数，然后从低位到高位依次相乘即可。时间复杂度为O(logn)。

为了避免低位很多零的情况，先过滤低位的0
源码如下：

template <class T, class Integer, class MonoidOperation>
T power(T x, Integer n, MonoidOperation op) {
  if (n == 0)
    return identity_element(op);
  else {
    while ((n & 1) == 0) {//过滤低位的0
      n >>= 1;
      x = op(x, x);
    }

    T result = x;
    n >>= 1;            //右移动1位，相当于除以2
    while (n != 0) {
      x = op(x, x); 
      if ((n & 1) != 0) //最后一位是1，是0就不用乘啦！
        result = op(result, x);
      n >>= 1;         //计算下一位
    }
    return result;
  }
}