数论之模加法运算

问题

一个$m$位的整数，有多少可以被整数$n$整除？其中$m$可以大于15。

• 测试输入
  1 3
• 测试输出
  4

分析1

这个问题显然不能穷举所有的情况。数论中的模加法运算有这样的性质（对乘法也是一样）：

$$\begin{array}{lcl} (a+b)\ \textrm{mod}\ n & = & (a\ \textrm{mod}\ n+b\ \textrm{mod}\ n)\ \textrm{mod}\ n\\ & = & (a\ \textrm{mod}\ n+b)\ \textrm{mod}\ n\\ & = & (a+b\ \textrm{mod}\ n)\ \textrm{mod}\ n\\ \end{array}$$

由此可见，一个数的取模问题可以拆分成两个求和数的取模问题，这显然属于动态规划能够解决的问题。对于一个$m$位的整数$x$，可以将其按照十进制拆解为$x=x_{0}×10^{m-1}+x_{1}×10^{m-2}+\cdots +x_{m-2}×10+x_{m-1}$。对于其中的$i-1$和$i$位做如下分析：
令$j=x_{i-1}\ \textrm{mod}\ n$，则

$$\begin{array}{lcl} (x_{i-1}×10+x_{i})\ \textrm{mod}\ n & = & [(x_{i-1}×10)\ \textrm{mod}\ n+x_{i}]\ \textrm{mod}\ n\\ & = & \{[(x_{i-1}\ \textrm{mod}\ n)×10]\ \textrm{mod}\ n+x_{i}\}\ \textrm{mod}\ n\\ & = & [(j×10)\ \textrm{mod}\ n+x_{i}]\ \textrm{mod}\ n\\ & = & (j×10+x_{i})\ \textrm{mod}\ n \end{array}$$

由此定义$c[i,j]$为前$i$位数模$n$余数为$j$的个数，原问题为$c[m,0]$。第$i$位的取值$k=0\sim 9$，穷举一遍后前$i$位余数为$(j*10+k)\%n$的个数，等于前$i-1$位余数为$j$的个数的累计和。从而递归式为

$$\left\{\begin{array}{ll} c[0,0]=1,\quad c[i,j]=0, & i=0,j\neq 0\\ c[i,(j*10+k)\%n]\ +=c[i-1,j], & i\neq 0, k=0\sim9 \end{array} \right.$$

代码1

import java.util.Scanner;

public class ModNums {
    static void solution(long[][] c, int m, int n) {
        for (int i = 0; i < n; i++) {
            c[0][i] = 0L;
        }
        c[0][0] = 1L;
        for (int i = 1; i <= m; i++) {
            for (int j = 0; j < n; j++) {
                for (int k = 0; k < 10; k++) {
                    c[i][(j * 10 + k) % n] += c[i - 1][j];
                }
            }
        }
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int m = sc.nextInt();
        int n = sc.nextInt();
        long[][] c = new long[m + 1][n];
        solution(c, m, n);
        System.out.println(c[m][0]);
    }
}

变形

一个$m$位的整数，其中有某些位不确定，用”X”表示，有多少可以被整数$n$整除？

• 测试输入
  999999999X 3
• 测试输出
  4

分析2

整体和原问题一样，只是不确定的位才需要穷举一遍，确定的位直接更新即可。

代码2

import java.util.Scanner;

public class AmbiguousNums {
    static void solution(long[][] c, String seq, int n) {
        int length = seq.length();
        for (int i = 0; i < n; i++) {
            c[0][i] = 0L;
        }
        c[0][0] = 1L;
        for (int i = 1; i <= length; i++) {
            char ch = seq.charAt(i - 1);
            for (int j = 0; j < n; j++) {
                if (ch == 'X') {
                    for (int k = 0; k < 10; k++) {
                        c[i][(j * 10 + k) % n] += c[i - 1][j];
                    }
                } else {
                    int k = ch - '0';
                    c[i][(j * 10 + k) % n] += c[i - 1][j];
                }
            }
        }
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String seq = sc.next();
        int length = seq.length();
        int n = sc.nextInt();
        long[][] c = new long[length + 1][n];
        solution(c, seq, n);
        System.out.println(c[length][0]);
    }
}