快速幂 -leetcode 1922. 统计好数字的数目

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\ast 3{)}^5=9^5=9\ast (9\ast 9{)}^2=9\ast 81^2=9\ast (81\ast 81{)}^1=9\ast 6561\ast 81^0=9\ast 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)}\ast 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;
    }
}