POJ1840---公式

题目描述

考虑具有以下形式的方程：
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;
}