蓝桥杯 四平方和【第七届】【省赛】【A组】暴力枚举 C++

资源限制

时间限制：1.0s 内存限制：256.0MB

　　四平方和定理，又称为拉格朗日定理：
　　每个正整数都可以表示为至多4个正整数的平方和。
　　如果把0包括进去，就正好可以表示为4个数的平方和。

　　比如：
　　5 = 0^2 + 0^2 + 1^2 + 2^2
　　7 = 1^2 + 1^2 + 1^2 + 2^2
　　（^符号表示乘方的意思）

　　对于一个给定的正整数，可能存在多种平方和的表示法。
　　要求你对4个数排序：
　　0 <= a <= b <= c <= d
　　并对所有的可能表示法按 a,b,c,d 为联合主键升序排列，最后输出第一个表示法


　　程序输入为一个正整数N (N<5000000)
　　要求输出4个非负整数，按从小到大排序，中间用空格分开

　　例如，输入：
　　5
　　则程序应该输出：
　　0 0 1 2

　　再例如，输入：
　　12
　　则程序应该输出：
　　0 2 2 2

　　再例如，输入：
　　773535
　　则程序应该输出：
　　1 1 267 838

　　资源约定：
　　峰值内存消耗（含虚拟机） < 256M
　　CPU消耗 < 3000ms


　　请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

　　所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
　　注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。
　　注意：主类的名字必须是：Main，否则按无效代码处理。

思路

枚举也要学会略去无效枚举。此处如果直接利用四重循环枚举，显然会TLE。稍加一点处理，便可轻松AC此题。

法一：

枚举a,b,c。而d由a,b,c确定。并且枚举，a从0开始，b从a开始，c从b开始。保证了a<=b<=c。

则b=sqrt(n-a*a-b*b-c*c)。当此b满足要求时，不用判断是否d>=c。证明：假设算出来的d1=<c1,那么枚举c时，c的值一定是经过了此d1的值，那么当c=c2=d1时，算出来d2便=c1,则d2=c1>c2=d1。所以第一次出现的d一定>=c。

如此枚举下来，找到的结果便是唯一确定且是最大数集。直接输出便可

#include<bits/stdc++.h>
using namespace std;
int n, a, b, c, d;
int main()
{
	scanf("%d", &n);
	for (a = 0; a *a<=n; ++a){//枚举到sqrt(n)便可，但是a*a<=n比用sqrt更快
		for (b = a; a*a+b *b<= n; ++b){
			for (c = b; a*a+b*b+c *c<=n; ++c){	//枚举前三个，第四个数不用枚举，直接由前三个数确定。
				d = sqrt(n - a * a - b * b - c * c);
				if (n == a * a + b * b + c * c + d * d){//说明此d合法
					//可直接输出，是因为如此枚举顺序保证了d>=c>=b>=a。只要找到了，那便是最大的合法数集
					printf("%d %d %d %d", a, b, c, d);
					return 0;
				}
			}
		}
	}
}

 

法二：

预处理计算c和d的平方和，存储在数组中。再枚举a和d，查找数组中是否存在数

t=n-a*a-b*b。存在即可直接输出，原理同法一。

#include<bits/stdc++.h>
using namespace std;
int c1[5000005], d1[5000005];//分别储存构成数i的c，d的数值

int main()
{
    int n, a, b, c, d, t;//放在局部更快
    scanf("%d", &n);
    memset(c1, -1, sizeof(c1));//初始化为-1
    for (c = 0; c * c <= n; c++) {
        for (d = c; d * d + c * c <= n; d++){
            t = c * c + d * d;
            if (c1[t] == -1){//说明首次出现
                c1[t] = c;
                d1[t] = d;
            }
        }
    }

    for (a = 0; a * a <= n; a++) {
        for (b = a; b * b + a * a <= n; b++){
            if (c1[n - a * a - b * b] != -1){//查找成功
                printf("%d %d %d %d", a, b, c1[n - a * a - b * b], d1[n - a * a - b * b]);
                return 0;
            }
        }
    }
    return 0;
}