【Leetcode 每日一题】1922. 统计好数字的数目

问题背景

我们称一个数字字符串是 好数字 当它满足（下标从 $0$ 开始）偶数 下标处的数字为 偶数 且 奇数 下标处的数字为 质数 $(2,3,5$ 或 $7)$。

• 比方说，“2582” 是好数字，因为偶数下标处的数字 $(2$ 和 $8)$ 是偶数且奇数下标处的数字 $(5$ 和 $2)$ 为质数。但 “3245” 不是好数字，因为 $3$ 在偶数下标处但不是偶数。

给你一个整数 $n$，请你返回长度为 $n$ 且为好数字的数字字符串 总数 。由于答案可能会很大，请你将它对 $(10^9+7)$ 取余后返回 。
一个 数字字符串 是每一位都由 $0$ 到 $9$ 组成的字符串，且可能包含前导 $0$。

数据约束

• $1\le n\le 10^{15}$

解题过程

注意题目中的 $n$ 表示数据的长度而不是范围，对长度可能达到 $10^{15}$ 量级的数字，用暴力解是一定会超时的。
给定的数字中允许包含前导 $0$，实际上意味着每个整数奇数位上的数字和偶数位上的数字可能性是确定的，这是一个纯粹的排列组合问题。
考虑到数据范围特别大，结果需要用 快速幂 来计算。

具体实现

class Solution {
    private static final int MOD = 1000000007;

    public int countGoodNumbers(long n) {
        return (int) (pow(5, (n + 1) >> 1) * pow(4, n >> 1) % MOD);
    }

    private long pow (long x, long n) {
        long res = 1;
        while (n > 0) {
            if ((n & 1) > 0) {
                res = res * x % MOD;
            }
            x = x * x % MOD;
            n >>= 1;
        }
        return res;
    }
}