方程的解数
问题描述
已知一个n元高次方程:
其中:x1, x2, …,xn是未知数,k1,k2,…,kn是系数,p1,p2,…pn是指数。且方程中的所有数均为整数。
假设未知数1≤ xi ≤M, i=1,,,n,求这个方程的整数解的个数。
输入文件(equation.in)
文件的第1行包含一个整数n。第2行包含一个整数M。第3行到第n+2行,每行包含两个整数,分别表示ki和pi。两个整数之间用一个空格隔开。第3行的数据对应i=1,第n+2行的数据对应i=n。
输出文件(equation.out)
文件仅一行,包含一个整数,表示方程的整数解的个数。
输入样例
3
150
1 2
-1 2
1 2
输出样例
178
约束条件
1<=n<=6,1<=M<=150;
方程的整数解的个数小于231。
★本题中,指数Pi(i=1,2,……,n)均为正整数。
#include<cstdlib>
#include<cmath>
#define max 4000037 //哈希函数中的大素数
using namespace std;
long answer[max],amount[max],use[max];//answer[i]是哈希表,amount[i]表示 s出现的次数,use[i]表示哈希表中该位置不位空
long mid;
long k[6],p[6];//k[i]是各项的系数,p[i]是各项的指数
long n,m,total;//n,m是题目中的约束条件,total是结果
long locate(long s)//哈希表的定位函数
{
long h;
h=s;
while (h<0) h+=max;
while (h>=max) h-=max;
while (use[h] && answer[h]!=s)
{
h++;
if (h>=max) h-=max;
}
return h;
}
void insert(long s)//哈希表的插入函数
{
long posi=locate(s);
if (!use[posi])
{ use[posi]=1;
answer[posi]=s;
}
amount[posi]++;
}
void firsthalf(long d,long s)//计算多项式前n/2项的和,保存在哈希表中
{
if (d==mid)
{
insert(s);
return;
}
long t;
for (long i=1; i<=m; i++)
{
t=k[d];
if (t!=0 && i!=1)
{
for (long j=0; j<p[d]; j++) t*=i;
}
firsthalf(d+1,t+s);
}
}
void find(long d,long s)//计算多项式后n/2项的和的相反数,并判断是否和前n/2项相等
{
if (d==n)
{
long posi;
s=-s;
posi=locate(s);
if (answer[posi]==s)
total+=amount[posi];
return;
}
long t;
for (long i=1; i<=m; i++)
{
t=k[d];
if (t!=0 && i!=1)
{
for (long j=0; j<p[d]; j++)
t*=i;
}
find(d+1,t+s);
}
}
int main()
{
freopen("dat.in","r",stdin);
freopen("dat.out","w",stdout);
cin>>n>>m;
for (long i=0; i<n; i++)
cin>>k[i]>>p[i];
mid=n/2;
firsthalf(0,0);
find(mid,0);
cout<<total<<endl;
}