POJ 1840 Eqs

如果直接枚举的话，复杂度为10^10

可以把方程拆成两半，一边三个未知数，另一边两个未知数，然后计算一边的所有可能值，用hash表储存，再枚举另一边的值，查找在哈希表中对应值出现的次数

 

这里有个优化，就是计算所有的可能值的时候，只计算2个数，因为如果计算3个数的话，会增加哈希表中的数据，导致查找过程变长

一个跑了375ms,另一个直接超时了。。。

 

另外注意xi != 0 !!!

 

代码：

#include<iostream> #include<cstdio> #include<memory.h> #define M 14997 struct node { int val; int next,cnt; }g[M*100]; int adj[M]; int a[6],e,ans; int find(int x,int va,int flag) { for(int i=adj[x];i!=-1;i=g[i].next) { if(g[i].val==va) { if(flag==1) g[i].cnt++; else ans+=g[i].cnt; return 1; } } return 0; } void add(int x,int val) { g[e].val=val; g[e].cnt=1; g[e].next=adj[x]; adj[x]=e++; } int main() { int i,j,k,t,r,p; e=0; ans=0; memset(adj,-1,sizeof(adj)); memset(g,0,sizeof(g)); for(i=1;i<=5;i++) scanf("%d",&a[i]); for(i=-50;i<=50;i++) { if(i==0) continue; for(j=-50;j<=50;j++) { if(j==0) continue; t=-a[4]*i*i*i-a[5]*j*j*j; if(t<0) r=(0-t)%M; else r=t%M; p=find(r,t,1); if(!p) add(r,t); } } for(i=-50;i<=50;i++) { if(i==0) continue; for(j=-50;j<=50;j++) { if(j==0) continue; for(k=-50;k<=50;k++) { if(k==0) continue; t=a[1]*i*i*i+a[2]*j*j*j+a[3]*k*k*k; if(t<0) r=(0-t)%M; else r=t%M; p=find(r,t,2); } } } printf("%d/n",ans); return 0; }