题意是给出一个五元三次方程组的五个系数
求共有多少组解
题目限制时间是5000Ms
如果直接用暴力求解的话5层循环,每层100次,100的5次方,肯定不行
转换一下方程组
-(a*x1*x*x1+b*x2*x2*x2)=c*x3*x3*x3+d*x4*x4*x4+e*x5*x5*x5
这样就降到了100的三次方,100万
然后我们用hash映射,将前面的结果算出来映射到hash表里,因为有可能是负值,我们将数组大小扩大到2倍也就是 2*(50*50*50*50*2)
如果是负值,就加上25000000
代码如下
#include<iostream>
#include<cstdio>
#include<cstring>
#define MAXN 25000002
using namespace std;
int a,b,c,d,e;
short hash[MAXN];
int answer;
int main()
{
while(cin>>a>>b>>c>>d>>e)
{
memset(hash,0,sizeof(hash));
//打表
for(int i=-50;i<=50;++i)
{
if(i)
{
for(int j=-50;j<=50;++j)
{
if(j)
{
int temp=-(a*i*i*i+b*j*j*j);
if(temp<0)
temp+=25000000;
hash[temp]++;
}
}
}
}
for(int i=-50;i<=50;++i)
{
if(i)
{
for(int j=-50;j<=50;++j)
{
if(j)
{
for(int k=-50;k<=50;++k)
{
if(k)
{
int temp = c*i*i*i+d*j*j*j+e*k*k*k;
if(temp<0)
temp+=25000000;
answer+=hash[temp];
}
}
}
}
}
}
cout<<answer<<endl;
}
}