2016蓝桥杯省赛C/C++A组第八题 四平方和

时间:2022-09-10 13:48:34

题意:

四平方和定理,又称为拉格朗日定理: 
每个正整数都可以表示为至多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个非负整数,按从小到大排序,中间用空格分开

例如,输入: 

则程序应该输出: 
0 0 1 2 

分析:直接暴力,三层循环。

#include<bits/stdc++.h>
using namespace std;
int main(){
    int N;
    scanf("%d", &N);
    bool ok = false;
    for(int a = 0; a * a <= N; ++a){
        for(int b = 0; b * b <= N; ++b){
            for(int c = 0; c * c <= N; ++c){
                int tmp = N - a * a - b * b - c * c;
                double t1 = (double)tmp;
                int t = (int)sqrt(t1);
                if(t * t == tmp){
                    printf("%d %d %d %d\n", a, b, c, t);
                    ok = true;
                    break;
                }
            }
            if(ok) break;
        }
        if(ok) break;
    }
    return 0;
}