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);
}
}
}
}
}
}