蒜头君在求解一个 nn 元的高次方程:
\displaystyle k_1x_1^{p_1}+k_2x_2^{p_2}+\ldots+k_nx_n^{p_n}=0k1x1p1+k2x2p2+…+knxnpn=0
其中:x_1,x_2,\ldots,x_nx1,x2,…,xn 是未知数,k_1,k_2,\ldots,k_nk1,k2,…,kn 是系数,p_1,p_2,\ldots,p_np1,p2,…,pn 是指数。方程中所有数都一定是整数。
假设未知数 1\leq x_i\leq M,i=1 \ldots n1≤xi≤M,i=1…n。你能帮蒜头君算出这个方程的整数解个数吗?
输入格式
第一行输入一个整数 n(1 \leq n \leq 4)n(1≤n≤4)。
第二行输入一个整数 M(1 \leq M \leq 150)M(1≤M≤150)。
第 33 行到第 n+2n+2 行,每行输入两个整数,分别表示 k_i(|k_i| \leq 20)ki(∣ki∣≤20) 和 p_i(1 \leq p_i \leq 4)pi(1≤pi≤4)。两个整数之间用一个空格隔开。
输出格式
输出一行,输出一个整数,表示方程的整数解的个数。
#include <bits/stdc++.h>
using namespace std;
int n,m;
int k[5];
int p[5];
int _pow[160][5];
int ans=0;
void in(){
memset(_pow,1,sizeof(_pow));
for(int i=1;i<=150;i++)
for(int j=1;j<=4;j++)
_pow[i][j]=i*_pow[i][j-1];
}
void dfs(int num,int mark){
if(num==n){
if(mark==0){
ans++;
}
return ;
}
for(int i=1;i<=m;i++){
dfs(num+1,mark+k[num]*_pow[i][p[num]]);
}
}
int main(){
cin>>n;
cin>>m;
in();
for(int i=0;i<n;i++)
cin>>k[i]>>p[i];
dfs(0,0);
cout<<ans<<endl;;
}