leetcode 周赛 248 场,第三题 1922. 统计好数字的数目
这个题其实就是求快速幂,但是一开始没看数据范围(n 为 1e15),直接 for 模拟,时间复杂度 O(n),超时了。
// tle
class Solution {
public int countGoodNumbers(long n) {
long ans = 1;
for (long i = 0L; i < n; i++) {
if (i % 2 == 0) {
ans = (long) ((ans * 5L) % (1e9 + 7)); // 这里其实也不对,1e9 + 7 应强转为 long
} else {
ans = (long) ((ans * 4L) % (1e9 + 7)); // 同上,应强转为 long
}
}
return (int) ans;
}
}
快速幂:
参考博客:快速幂算法
快速幂思想:
- 每一步都把指数分成两半,而相应的底数做平方运算。
- 指数为偶数时,指数直接减半即可;
- 指数为奇数时,需要先拿出来一个,将指数变成偶数,然后指数再减半。
- 最终使得 for 幂运算的 O(n)时间复杂度降为 O(log n)
例如, 3 10 = ( 3 ∗ 3 ) 5 = 9 5 = 9 ∗ ( 9 ∗ 9 ) 2 = 9 ∗ 8 1 2 = 9 ∗ ( 81 ∗ 81 ) 1 = 9 ∗ 6561 ∗ 8 1 0 = 9 ∗ 6561 = 59049 3^{10} = (3*3)^5 = 9^5 = 9 * (9*9)^2 = 9*81^2 = 9*(81*81)^1 = 9*6561*81^0 = 9*6561 = 59049 310=(3∗3)5=95=9∗(9∗9)2=9∗812=9∗(81∗81)1=9∗6561∗810=9∗6561=59049
class Solution {
public int countGoodNumbers(long n) {
long i = n / 2L; // 偶数个数
long j = n / 2L; // 奇数个数
long mod = (long) (1e9 + 7L);
if (n % 2 == 1) { // 偶数多一个
i++;
}
return (int) ((pow(5L, i, mod) * pow(4L, j, mod)) % mod);
}
/**
* 快速幂
* @param n 底数
* @param m 次幂
* @return
*/
static long pow(long n, long m, long mod) {
long res = 1L;
long base = n; // 底数
while(m != 0) { // 指数为 0 停止
if ((m & 1) == 1) { // 次幂为奇数(等价于 m % 2 == 1,但是位运算能加快速度)
res = (res * base) % mod;
}
base = (base * base) % mod;
m = m >> 1; // 相当于 m = m / 2; 但是,位运算更快
}
return res;
}
}
这个题,主要就是根据 n 然后,求一下 4 ( n / 2 ) ∗ 5 ( n / 2 ) 4^{(n / 2)} * 5 ^ {(n / 2)} 4(n/2)∗5(n/2) %mod;
但是,比赛时候 没有把 mod 指定为 long 类型,所以出错了…
还是基础不牢啊,long 和 int 转换竟然不清楚…
- int to long,不会自动转换,也要强转(整型默认为 int)
- long to int 要强转
50. Pow(x, n)
题目描述
实现 pow(x, n) ,即计算 x 的 n 次幂函数(即,xn)。
经典快速幂问题。
但是,一开始踩坑了。
注意:
- 这里
power 必须是 long
;否则,如果是int的话,Math.abs(-2147483648) =-2147483648 会 int 溢出
其中,Math.abs
源码如下:
所以,power 和 n 一定要为 long
class Solution {
public double myPow(double x, int n) {
if (n == 0) {
return 1.0;
}
double res = 1.0;
double base = x;
int power = Math.abs(n); // 指数取正
while (power > 0) {
if ((power & 1) == 1) {
res *= base;
}
power /= 2;
base *= base;
}
// 如果指数为负数,则结果为 正指数幂的 倒数
if (n < 0) {
System.out.println("n < 0");
res = 1.0 / res;
}
return res;
}
}