题目描述
考虑具有以下形式的方程:
a1x13+ a2x23+ a3x33+ a4x43+ a5x53=0
系数被赋予区间 [-50,50] 中的整数。
考虑一个解一个系统 (x1, x2, x3, x4, x5) 来验证方程,xi∈[-50,50], xi != 0, any i∈{1,2,3,4,5 }.
确定有多少解满足给定的方程。
输入
输入的唯一一行包含 5 个系数 a1、a2、a3、a4、a5,以空格分隔。
输出
输出将在第一行包含给定方程的解数。
样本输入
37 29 41 43 47
样本输出
654
思路分析:直接暴力必定超时,我们可以对方程进行变形.
这样100^5的复杂度就变成了100 ^3 + 100 ^ 2.
然后将等式从左或右的值枚举存入哈希表,由于可能存在负值我们让负值+25000000转为正数,并且保证数值的唯一性.之后再暴力枚举另一边,将哈希表对应的值存入ans进行累加,最后得到ans的值.
并且25000000数组很大,int不可以,short足够,并且要定义成全局.
AC代码
//POJ1840
#include<iostream>
#include<string.h>
using namespace std;
const int maxsize = 25000000+10;//
short hashTab[maxsize];
int a1,a2,a3,a4,a5,ans,temp;
int main()
{
while(cin>>a1>>a2>>a3>>a4>>a5){
ans = 0;
memset(hashTab,0,sizeof(hashTab));
for(int x1 = -50;x1<=50;x1++){
for(int x2 = -50;x2<=50;x2++){
if(x1==0||x2==0){
continue;
}
temp = (a1*x1*x1*x1+a2*x2*x2*x2)*(-1);
if(temp < 0){
temp += maxsize;
}
hashTab[temp]++;
}
}
for(int x3 = -50;x3<=50;x3++){
for(int x4 = -50;x4<=50;x4++){
for(int x5 = -50;x5<=50;x5++){
if(x3==0||x4==0||x5==0){
continue;
}
temp = a3*x3*x3*x3+a4*x4*x4*x4+a5*x5*x5*x5;
if(temp < 0){
temp += maxsize;
}
ans += hashTab[temp];
}
}
}
cout<<ans<<endl;
}
return 0;
}