POJ 2810 完美立方（枚举，二分查找binary_search的运用）

POJ 2810 完美立方

题目描述：

a的立方 = b的立方 + c的立方 + d的立方为完美立方等式。例如12的立方 = 6的立方 + 8的立方 + 10的立方 。编写一个程序，对任给的正整数N (N≤100)，寻找所有的四元组(a, b, c, d)，使得a的立方 = b的立方 + c的立方 + d的立方，其中a,b,c,d 大于 1, 小于等于N。

Input

正整数N (N≤100)

Output

每行输出一个完美立方，按照a的值，从小到大依次输出。当两个完美立方等式中a的值相同，则依次按照b、c、d进行非降升序排列输出，即b值小的先输出、然后c值小的先输出、然后d值小的先输出。

Sample Input

24

Sample Output

Cube = 6, Triple = (3,4,5)
Cube = 12, Triple = (6,8,10)
Cube = 18, Triple = (2,12,16)
Cube = 18, Triple = (9,12,15)
Cube = 19, Triple = (3,10,18)
Cube = 20, Triple = (7,14,17)
Cube = 24, Triple = (12,16,20)

思路：

现将到N的三次方的数存储起来（避免重复计算），然后按照题目要求进行枚举查找符合条件a,b,c,d 的值,满足要求输出。
四层循环太复杂，两种做法：后面有降低时间复杂度N^3的写法：binary_search（二分查找）

实现：

#include<cstdio>
#include<string>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int main()
{
	int N;
	cin>>N;
	int a,b,c,d,cube[3015];
	for(int i=2;i<=N;i++)cube[i]=i*i*i;
	for(a=6;a<=N;a++)
	for(b=2;b<a;b++)
	for(c=b;c<a;c++)
	for(d=c;d<a;d++)
		if(cube[a]==cube[b]+cube[c]+cube[d])cout<<"Cube = "<<a<<","<<" Triple = ("<<b<<","<<c<<","<<d<<")"<<endl;
}

下面改用二分查找缩短运算时间：

#include<cstdio>
#include<string>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int main()
{
	int N,a,b,c,d;
	int A[3015];
	scanf("%d",&N);
	for(int i=2;i<=N;i++)
	A[i]=i*i*i;
	for(a=6;a<=N;a++)
	{
		for(b=2;b<a;b++)
		{
			for(c=b;c<a;c++)
			{
				int tp=(int)(pow(a,3)-pow(b,3)-pow(c,3));
				if(binary_search(A+2,A+N+1,tp))
				{
					for(int i=2;i<=N;i++)
					{
						if(A[i]==tp&&i>c)
						printf("Cube = %d, Triple = (%d,%d,%d)\n",a,b,c,i);
					}
				}
			}
		}
	}
}