这题也能用hash做,涨姿势了……
思路:将等式变形为a*x1^2+b*x2^2= -(c*x3^2+d*x4^2) 先用两重循环列举a,b的所有情况,将等式的左边结果存入hash表。再用两重循环列举c,d的所有情况,看看结果的相反数在不在hash表中。统计输出。
优化:x1^2与x1的正负号无关,x2--x4同理,循环时从1-100循环即可,最后答案乘以16;若abcd均大于或小于0,直接输出0.
HDU ACM的PPT上有两种解法。第一种用了大数组,第二种用了小数组+冲突处理。
由于小数组的初始化时间少,第二种解法更优。
#include
#include
using namespace std;
int a,b,c,d,pin[101],hash[2000010],i,j,sum;
int main()
{
for (i=1;i<=100;++i)
pin[i]=i*i;
while (cin>>a>>b>>c>>d){
if ((a>0 && b>0 && c>0 && d>0) || (a<0 && b<0 && c<0 && d<0)){
cout<<"0\n";
continue;
}
memset(hash,0,sizeof(hash)); sum=0;
for (i=1;i<=100;++i)
for (j=1;j<=100;++j)
hash[a*pin[i]+b*pin[j]+1000000]++;
for (i=1;i<=100;++i)
for (j=1;j<=100;++j)
sum+=hash[1000000-(c*pin[i]+d*pin[j])];
cout<
#include
#include
using namespace std;
const int MAX=50021;
int f[MAX],g[MAX];
int a,b,c,d,pin[101],i,j,sum,s,p;
int hash(int k){
int t=k%MAX;
if (t<0) t+=MAX;
while (f[t]!=0 && g[t]!=k)
t=(t+1)%MAX;
return t;
}
int main()
{
for (i=1;i<=100;++i)
pin[i]=i*i;
while (cin>>a>>b>>c>>d){
if ((a>0 && b>0 && c>0 && d>0) || (a<0 && b<0 && c<0 && d<0)){
cout<<"0\n";
continue;
}
memset(f,0,sizeof(f)); sum=0;
for (i=1;i<=100;++i)
for (j=1;j<=100;++j){
s=a*pin[i]+b*pin[j];
p=hash(s);
g[p]=s;
f[p]++;
}
for (i=1;i<=100;++i)
for (j=1;j<=100;++j){
s=-(c*pin[i]+d*pin[j]);
p=hash(s);
sum+=f[p];
}
cout<<(sum<<4)<
kdwycz的网站: http://kdwycz.com/
kdwyz的刷题空间:http://blog.csdn.net/kdwycz