POJ 1840 Eqs(hash表)

Description
给出一个5元3次方程a1*x1^3+a2*x2^3+a3*x3^3+a4*x4^3+a5*x5^3=0,输入其5个系数,求它的解的个数,其中系数 ai∈[-50,50] 自变量xi∈[-50,0)∪(0,50]
Input
方程的五个系数
Output
方程解的个数
Sample Input
37 29 41 43 47
Sample Output
654
Solution
转化为ax1^3+bx2^3+cx3^3=-dx4^3-ex5^3,利用哈希查找的方式优化枚举,我先将方程左边的式子的值打表存放,然后枚举右边的式子,对于右边每一种计算的可能,判断哈希位置在哈希表中是否存在,若不存在,则说明左边所有式子的可能取值中没有一个与其相等。若存在,则说明在该左边式子所有可能的取值中只有在该哈希位置的值才可能与其相等。接下来就是逐个比较,若相等则count++,其中count为记录方程所有可能解的个数。否则比较下一个。
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define INF (1<<29)
#define maxn 2000000
int a,b,c,d,e;
int hash[maxn];
int ans;
int main()
{
    scanf("%d%d%d%d%d",&a,&b,&c,&d,&e);
    int res=0;
    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 temp=i*i*i*a+j*j*j*b+k*k*k*c;
                hash[res++]=temp;//打hash表 
            }
        }
    }
    sort(hash,hash+res);//对hash表排序 
    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 temp=i*i*i*d+j*j*j*e;
            ans+=upper_bound(hash,hash+res,-temp)-lower_bound(hash,hash+res,-temp);//记录解的数量 
        }
    }
    printf("%d\n",ans);
    return 0;
}

你可能感兴趣的:(POJ 1840 Eqs(hash表))