计蒜客：方程的解数（java）

dfs算法

蒜头君在求解一个n元的高次方程：

 

其中：x1,x2,…,xn 是未知数，k1,k2,…,kn是系数，p1,p2,…,pn是指数。方程中所有数都一定是整数。

假设未知数 1≤xi≤M,i=1…n。你能帮蒜头君算出这个方程的整数解个数吗？

输入格式

第一行输入一个整数 n(1≤n≤4)。

第二行输入一个整数 M(1≤M≤150)。

第3行到第 n+2 行，每行输入两个整数，分别表示 ki(∣ki∣≤20)和 pi(1≤pi≤4)。两个整数之间用一个空格隔开。

输出格式

输出一行，输出一个整数，表示方程的整数解的个数。

样例输入

3

100

1 2

-1 2

1 2

样例输出

104

public class Main {
    static int ans = 0, n, m;
    static int[] a = new int[4];
    static int[] b = new int[4];

    static int quick_pow(int x, int k) {
        int res = 1;
        while (k > 0) {
            if ((k & 1) == 1) res = res * x;
            x *= x;
            k >>= 1;
        }
        return res;
    }

    static void dfs(int u, int sum) {
        if (u >= n) {
            if (sum == 0) ans+=1;
            return;
        }
        for (int i = 1; i <= m; i++) {
            int k = a[u], p = b[u];
            int c = k * quick_pow(i, p);
            dfs(u + 1, sum + c);
        }
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        n = scanner.nextInt();
        m = scanner.nextInt();
        for (int i = 0; i < n; i++) {
            int a1, b1;
            a1 = scanner.nextInt();
            b1 = scanner.nextInt();
            b[i] = b1;
            a[i] = a1;
        }
        dfs(0, 0);
        System.out.println(ans);
    }
}