AIDreamer

复杂性科学 机器学习 ACM 思维认知 https://moumuyun.wordpress.com/

POJ 1840 Eqs(Hash+思维)

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();
    }
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/mmy1996/article/details/52374030
文章标签: hash poj
个人分类: 哈希
上一篇从神经元连接的角度谈谈思维
下一篇POJ 2503 Babelfish(map或hash或排序二分)
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

关闭
关闭