HDU 1496 Equations

题意:求方程解的个数

这里4个x都不相关,所以只要计算 ax1^2+bx2^2==cx3^2+dx4^2

遍历找出左边所有的解,然后对应找出右式的值对应的答案(这里0都挖掉了)

从1开始遍历(因为平方-1和1解不重复计算)最后结果*16 //这一步优化很给力


#include <stdio.h>
#include <string.h>
#define N 1000001
int hash1[N],hash2[N];
int main(){
	int a,b,c,d,i,j;
	while(~scanf("%d%d %d %d",&a,&b,&c,&d))
	{
		if((a>0&&b>0&&c>0&&d>0) || (a<0&&b<0&&c<0&&d<0)){printf("0\n");continue;}
		memset(hash1,0,sizeof(hash1));
		memset(hash2,0,sizeof(hash2));
		for(i=-100;i<=100;i++)
			for(j=-100;j<=100;j++)
			{
				if(i==0 ||j==0)continue;
				int k=a*i*i+b*j*j;
				if(k>=0)hash1[k]++;
				else hash2[-k]++;
			}
			int ans=0;
			for(i=-100;i<=100;i++)
				for(j=-100;j<=100;j++)
				{
					if(i==0 ||j==0)continue;
					int k=c*i*i+d*j*j;
					if(k>0)	ans+=hash2[k];
					else    ans+=hash1[-k];
				}
		printf("%d\n",16*ans);
	}return 0;
}


你可能感兴趣的:(HDU 1496 Equations)