题意:
给出ai(i=1~5),求a1 * x1^3+ a2 * x2^3+ a3 * x3^3+ a4 * x4^3+ a5 * x5^3=0在-50到50之间的x的解的个数
分析:
把前三个的和放到hash里,然后枚举后两个的和,查一下hash即可。
这题也可以用map去做。
#include<iostream> #include<cstdio> #include<vector> #include<cstring> #include<algorithm> #include<map> using namespace std; const int prime=997; const int l=-50,r=50; vector<int>hs[prime]; int main() { int a[6]; for(int i=1;i<6;i++)scanf("%d",&a[i]); for(int i=l;i<=r;i++) for(int j=l;j<=r;j++) for(int k=l;k<=r;k++){ if(!i||!j||!k)continue; int t=i*i*i*a[1]+j*j*j*a[2]+k*k*k*a[3]; hs[abs(t)%prime].push_back(t); } int ans=0; for(int i=l;i<=r;i++) for(int j=l;j<=r;j++){ if(!i||!j)continue; int t=-(i*i*i*a[4]+j*j*j*a[5]); int id=abs(t)%prime; for(int k=0;k<hs[id].size();k++) if(hs[id][k]==t)ans++; } printf("%d ",ans); return 0; }