关闭

POJ 1840 Eqs(Hash+思维)

标签: hashpoj
118人阅读 评论(0) 收藏 举报
分类:

poj 1840

题目大意

求方程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();
    }
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:70096次
    • 积分:2361
    • 等级:
    • 排名:第17428名
    • 原创:167篇
    • 转载:7篇
    • 译文:0篇
    • 评论:147条
    最新评论
    链接