问题描述
蒜头君在求解一个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
#include <stdio.h>
#include <iostream>
#include <cstring>
using namespace std;
int n,M;
int k[5],p[5];
int mypow[160][5];
int ans=0;
void init()
{
memset(mypow,1,sizeof(mypow));
for(int i=1;i<=150;i++)
for(int j=1;j<=4;j++)
mypow[i][j]=i*mypow[i][j-1];
}
void dfs(int num,int mark)
{
if(num!=0&&num!=n&&mark==0)
return;
if(num==n)
{
if(mark==0)
ans++;
return;
}
for(int i=1;i<=M;i++)
dfs(num+1,mark+k[num]*mypow[i][p[num]]);
}
int main()
{
cin>>n;
cin>>M;
init();
for(int i=0;i<n;i++)
cin>>k[i]>>p[i];
dfs(0,0);
cout<<ans;
return 0;
}