1. 程式人生 > >STL系列之七 快速計算x的n次冪 power 的實現

STL系列之七 快速計算x的n次冪 power 的實現

               

計算x的n次冪最簡單直接的方法就是相乘n次,很容易寫出程式:

//計算x^n 直接乘n次 by MoreWindows( http://blog.csdn.net/MoreWindows )int power1(int x, unsigned int n)int result = 1while (n--)  result *= x; return result;}

這種計算的效率顯然不高,我們可以用二分法來加速計算x^n=x^(n/2)* x^(n/2)即x^10=x^5*x^5,這種計算N次冪只要相乘O(logN)次。運用遞迴的方法不難寫出:

//計算x^n 二分遞迴實現  by MoreWindows( http://blog.csdn.net/MoreWindows )
int power2(int x, unsigned int n)if (n == 0)  return 1else if (n == 1)  return x; else  {  if (n % 2 == 1)   return power2(x, n / 2) * power2(x, n / 2) * x;  else   return power2(x, n / 2) * power2(x, n / 2); }}

遞迴畢竟比較浪費時間,且會有很多重複計算。

因此最好能換成非遞迴的方式來實現二分法。

考慮x^23,可以先從x ->x^2 -> x^4 -> x^8 -> x^16 取result1 = x^16,然後23-16=7。

我們只要計算x^7再與result1相乘就可以得到x^23。對於x^7也可以採用這種方法

取result2 = x^4,然後7-4=3,只要計算x^3再與result2相乘就可以得到x^7。由此可以將x^23寫成x^16 * x^4* x^2 * x,即23=16+4+2+1,而23 = 10111(二進位制),所以只要將n化為二進位制並由低位到高位依次判斷如果第i位為1,則result *=x^(2^i)。

函式實現如下:

//計算x^n   by MoreWindows( http://blog.csdn.net/MoreWindows )int power3(int x, unsigned int n)if
(n == 0)  return 1int result = 1while (n != 0) {  if ((n & 1) != 0)   result *= x;  x *= x;  n >>= 1; } return result;}

此函式可以在相乘O(logN)次內計算x的n次冪,且避免了重複計算。但還可以作進一步的優化,如像48=110000(二進位制)這種低位有很多0的數,可以先過濾掉低位的0再進行計算,這樣也會提高一些效率。程式如下:

//計算x^n  by MoreWindows( http://blog.csdn.net/MoreWindows )int power4(int x, unsigned int n)if (n == 0) {  return 1; } else {  while ((n & 1) == 0)  {   n >>= 1;   x *= x;  } } int result = x; n >>= 1while (n != 0) {   x *= x;  if ((n & 1) != 0)   result *= x;  n >>= 1; } return result;}

驗證一下

int main()printf("驗證power4()  -- by MoreWindows( http://blog.csdn.net/MoreWindows ) --\n\n"); for (int i = 0; i <= 10; i++)  printf("2的%d次方為\t%d\n", i, power4(2, i)); return 0;}

結果為

看到這裡,理解STL的power()函式應該就是個水到渠成的事情了——我們自己寫的power4()正是STL的power()函式。

 注,非常感謝網友evaxiao幫我找出了power4()的一個錯誤,我已經在文中改正了,謝謝網友evaxiao