ZSQ

新的一天,新的不会

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;
}
阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/V5ZSQ/article/details/46786767
个人分类: POJ hash
想对作者说点什么? 我来说一句

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

不良信息举报

POJ 1840 Eqs(hash表)

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭