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);
}
}