[蓝桥杯] 四平方和

[蓝桥杯] 四平方和

峰值内存消耗 < 256M　　CPU消耗  < 3000ms

【题目描述 - Problem Description】

四平方和定理，又称为拉格朗日定理： 每个正整数都可以表示为至多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	12	773535
输出	0 0 1 2	0 2 2 2	1 1 267 838

【题解】

　　搜索，注意剪枝。

　　四层循环的速度可以……因为强迫症自己又写了个DFS版本……

【代码 C++】

#include <cstdio>
int main() {
    int i1, i2, i3, i4, s1, s2, s3, s4, n;
    scanf("%d", &n);
    for (i1 = 0;; ++i1){
        s1 = i1*i1;
        for (i2 = i1; s1 + i2*i2 * 3 <= n; ++i2){
            s2 = s1 + i2*i2;
            for (i3 = i2; s2 + i3*i3 * 2 <= n; ++i3){
                s3 = s2 + i3*i3;
                for (i4 = i3; s3 + i4*i4 <= n; ++i4){
                    s4 = s3 + i4*i4;
                    if (s4 == n){
                        printf("%d %d %d %d
", i1, i2, i3, i4);
                        goto ed;
                    }
                }
            }
        }
    }
ed:;
    return 0;
}

#include <cstdio>
int n, q[5];
bool isFid;
void DFS(int lst, int i, int sum){
    if (isFid) return;
    if (i){
        int tmp;
        while (sum + lst*lst*i <= n){
            q[i] = lst;
            DFS(lst, i - 1, sum + lst*lst);
            ++lst;
        }
    }
    else if (sum == n){
        printf("%d %d %d %d
", q[4], q[3], q[2], q[1]);
        ++isFid;
    }
}
int main() {
    scanf("%d", &n);
    DFS(0, 4, 0);
    return 0;
}