給定整數 x (0 < x < 10^9) 、整數 n (0 <= n < 10^9),求取 x^n
Input #1: x = 2, n = 22
Output #1: 4194304
Input #2: x = 1, n = 0
Output #2: 1
通常這類題目還會給定一個數 M
(通常是 10^9+7
),然後要求輸出 (x ^ n) % M
。
因為這樣解題就不用使用大數了,這題要考的也不是大數。
快速冪是矩陣加速的基礎,矩陣加速是好物,所以盡量搞懂快速冪吧。
就把 x
乘 n
次,看是多少……時間複雜度: O(n)
當 n
的範圍大到 10^9,陣列是開不出來的。只能改用 map/dict 之類的資料結構來做,
這樣查詢的時間複雜度是 O(lg(MAX_N))
,還不錯,但預處理為 O(MAX_N * lg(MAX_N))
,
實在太花時間了,而且這樣做會需要極大的空間。
正式名稱叫 Exponentiation by squaring,不過許多人簡稱它為快速冪。
顯而易見的,我們可以在 O(lg(n))
的時間計算出
x, x^2, x^4, x^8, x^16, ..., x^(2^m) (2 ^ m <= n) ...... 式 1
這些數(每一項都是前一項的平方)。我們可以利用這些數 組合 出 x^n
。
例如
x^22 = (x^16) * (x^4) * (x^2)
x^9 = (x^8) * (x^1)
有沒有發現什麼?這根本就是 n
的二進位拆解啊!
沒看出來?以 x^22
為例
22 的二進位 | 1 | 0 | 1 | 1 | 0 |
---|---|---|---|---|---|
i | 4 | 3 | 2 | 1 | 0 |
2^i | 2^4 | 2^3 | 2^2 | 2^1 | 2^0 |
x^(2^i) | x^16 | x^8 | x^4 | x^2 | x^1 |
只要二進位中對應的位置 i
是 1,我們就要乘上對應的 x^(2^i)
。
於是,要計算 x^22
,我們只要計算出 (x^16)
, (x^4)
, (x^2)
。
而這些東西,我們知道可以快速地在 O(lg(n))
計算出。
於是,我們設計出一個演算法:
我們依序計算出
式 1
,針對第i
項,即x^(2^i)
,如果n
的二進位中的第i
項是1
的話,x^(2^i)
就是解的一部分。當我們把所有這種x^(2^i)
找出來後,全部乘起來就是x^n
了。
int fast_pow(int x, int n) {
int result = 1; // 解
int base = x; // x ^ (2^i)
for (int i = 0; (1 << i) <= n; base = base * base, i++) {
if (n & (1 << i)) { // 如果位置 i 是 1
result = result * base;
}
}
return result;
}
※ (1 << i)
就是 0000...001 這個數左移 i
位,變成 000...1...000,
讓它跟 n
做 AND(bitwise AND),即可知道 n
的二進位中第 i
位是不是 1。
回憶一下:
0 & 0 = 0
0 & 1 = 0
1 & 0 = 0
1 & 1 = 1
任何位元與 1 做 AND 就會得到它本身。
另一種寫法:
int fast_pow(int x, int n) {
int result = 1;
int base = x;
while (n != 0) { // 從尾端一個一個拆解出位元
if (n & 1)
result = result * base;
base = base * base;
n >>= 1; // 即 n /= 2
}
return result;
}
這個演算法還可以這麼想:
針對
x^n
,
如果
n
是偶數,則x^n = (x ^ 2) ^ (n / 2)
;
如果
n
是奇數,則x^n = (x ^ 2) ^ ((n-1) / 2) * x
;
用遞迴寫成程式:
int fast_pow(int x, int n) {
if (n == 0)
return 1;
if (n & 1) // n 是奇數,也可寫 if (n % 2 == 1)
return fast_pow(x * x, (n-1) / 2) * x;
else
return fast_pow(x * x, n / 2);
}
完全等價於前面用迴圈的寫法。
如果題目是要求,輸出的結果為 x^n mod M
,我們不可能把 x^n
計算出來後再 mod M
,
因為這樣很可能會在計算過程中,值超出 int 的範圍,所以利用
(a * b) mod M = ((a mod M) * (b mod M)) mod M
我們每次乘 base
到 result
時與 base
平方時,mod 一下, 最後的結果就是所求了。
int fast_pow(int x, int n, int M) {
int result = 1;
int base = x;
while (n != 0) { // 從尾端一個一個拆解出位元
if (n & 1)
result = result * base % M;
base = base * base % M;
n >>= 1; // 即 n /= 2
}
return result;
}
int fast_pow(int x, int n, int M) {
if (n == 0)
return 1;
if (n & 1) // n 是奇數,也可寫 if (n % 2 == 1)
return fast_pow((x * x) % M, (n-1) / 2) * x % M;
else
return fast_pow((x * x) % M, n / 2) % M;
}
實作時,通常 M
不會給太小,所以應全部使用 long long
,防止溢位。
基本上,只考快速冪的題目很少見,大多是混合著矩陣加速(矩陣快速冪)出題。
這裡放一些快速冪的題目:
Codeforces #324B: 這題測資較弱,用 naive solution 也可以解,不過可以用快速冪來寫寫看。這題主要是考組合數學。