POJ 1840 Eqs（Hash+思维）

poj 1840

题目大意

求方程$a_1x_1^3$+$a_2x_2^3$+ $a_3x_3^3$+ $a_4x_4^3$+ $a_5x_5^3$=0 解的个数，$a_i$∈[-50,50],$x_i$∈[-5-,50]且$x_i$!=0

分析

直接暴力枚举肯定超时，这里用到了一个巧妙的方法：

$$-(a_1x_1^3+a_2x_2^3)=a_3x_3^3+ a_4x_4^3+ a_5x_5^3$$
因为要判断多少组解满足上面等式，枚举左边将左边的值哈希后再枚举右边查找哈希表中是否存在右边的和，这样就能以O( $100^3+100^2$)的复杂度解决问题了。
这道题很考察思维的灵活性。

代码

#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();
    }
}