题目大意
求方程 a1x31 + a2x32 + a3x33 + a4x34 + a5x35 =0 解的个数, ai ∈[-50,50], xi ∈[-5-,50]且 xi !=0
分析
直接暴力枚举肯定超时,这里用到了一个巧妙的方法:
−(a1x31+a2x32)=a3x33+a4x34+a5x35
因为要判断多少组解满足上面等式,枚举左边将左边的值哈希后再枚举右边查找哈希表中是否存在右边的和,这样就能以O( 1003+1002 )的复杂度解决问题了。
这道题很考察思维的灵活性。
代码
#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
using namespace std;
const int INF=999999999;
int a1,a2,a3,a4,a5;
int Key=10007;
struct Hash
{
int sum;
int t;
}H[10010][50];
int cnt[10010];
bool Find(int x,int y)
{
int loc=abs(-a1*x*x*x-a2*y*y*y)%Key;
for(int i=1;i<=cnt[loc];i++)
{
if(H[loc][i].sum==(-a1*x*x*x-a2*y*y*y )){H[loc][i].t++;return 1;}
}
return 0;
}
void Work()
{
int ans=0;
for(int i=-50;i<=50;i++)
{
if(i==0)continue;
for(int j=-50;j<=50;j++)
{
if(j==0)continue;
int loc=abs(-a1*i*i*i-a2*j*j*j)%Key;
if(Find(i,j)==1)continue;
else
{
H[loc][++cnt[loc]].sum=-a1*i*i*i-a2*j*j*j;
H[loc][cnt[loc]].t=1;
}
}
}
for(int i=-50;i<=50;i++)
{
if(i==0)continue;
for(int j=-50;j<=50;j++)
{
if(j==0)continue;
for(int k=-50;k<=50;k++)
{
if(k==0)continue;
int loc=abs(a3*i*i*i+a4*j*j*j+a5*k*k*k)%Key;
//cout<<loc<<endl;
for(int p=1;p<=cnt[loc];p++)
{
if(H[loc][p].sum==(a3*i*i*i+a4*j*j*j+a5*k*k*k))ans+=H[loc][p].t;
}
}
}
}
cout<<ans<<endl;
}
int main()
{
while(scanf("%d%d%d%d%d",&a1,&a2,&a3,&a4,&a5)!=EOF)
{
memset(H,0,sizeof(H));
sizeof(cnt,0,sizeof(cnt));
Work();
}
}