hdu1496 equation(较好的hash)

初一看四个变量,四重循环,暴力肯定会超时,这个时候可以将等式移项一下,得到如下式子;

a*x1^2+b*x2^2=-c*x3^2-d*x4^2;这样一来,可以将左边的结果暴力枚举一下存在一个数组中,同时将右边的结果枚举一下,存在另一个数组,当两边结果相等时,用左边的方案数乘上右边的方案数就是最终的方案数;


#include
#include
#include
#include
#include
#include
#define N 2000100
int hash1[N],hash2[N];

int main()
{
    int a,b,c,d,i,j,ans;
    while(~scanf("%d%d%d%d",&a,&b,&c,&d))
    {
        if(a*b>0&&b*c>0&&c*d>0&&d*a>0)//如果abcd同正或者同负时候可以直接输出0,一定要判断,虽然不判断在后面运算中也能得到相同的结果,但是不判断会超时的。
        {
            printf("0\n");
            continue;
        }
        memset(hash1,0,sizeof(hash1));
        memset(hash2,0,sizeof(hash2));
        for(i=1;i<=100;i++)//由于解的平方一定为正数,可以只枚举解为正数的情况,然后每一个解都对应一正一负两个数,四个解一共可以有16种情况,所有在最后结果中乘上16;
        {
            for(j=1;j<=100;j++)
            {
                hash1[i*i*a+j*j*b+1000000]++;//因为最大的数是50*100*100+50*100*100等于1000000,所以加上1000000以防数组越界;
                hash2[1000000-i*i*c-j*j*d]++;
            }
        }
        ans=0;
        for(i=1;i<=N;i++)
        {
            ans+=hash1[i]*hash2[i];
        }
        printf("%d\n",16*ans);
    }
    return 0;
}



你可能感兴趣的:(基础题,hdu)