四平方和(蓝桥杯-暴力)

四平方和定理，又称为拉格朗日定理：每个正整数都可以表示为至多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

暴力做法：

#include <iostream>
#include <cmath>
using namespace std;


int main()
{
    int  n;
     cin>>n;
    //n=5000000;
    for(int a=0; a<=2237; a++)
    {
        for(int b=0; b<=2237; b++)
        {
            for(int c=0; c<=2237; c++)
            {
                int dj=n-a*a-b*b-c*c;
                int d=sqrt(dj);
                if(dj==d*d)//如果dj等于d的平方，第四位数是一个整数的平方，这个整数d就是第四位数
                {
                cout<<a<<" "<<b<<" "<<c<<" "<<d;
                return 0;
                }
            }
        }
    }
}

我的做法：

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
int f[3030]={0};
int dp[2000][5]={0},dp2[2500][2];
int main()
{
    for(int i=0;i<=3000;i++){
        f[i]=pow(i,2);
    }
    int n;
    scanf("%d",&n);
    int ans[4],cnt=0;
    for(int i=3000;i>=0;i--){
        if(n>=f[i]){
            int t=n-f[i],num=1,d=1;
            dp[cnt][0]=i;
            for(int j=i;j>=0;j--){
                if(num==4)break;
                if(t>=f[j]){
                    dp[cnt][num++]=j;
                    t-=f[j];
                    j++;
                }
            }
            cnt++;
        }
    }
    int ind=0,cnt2=0;
    for(int i=0;i<cnt;i++){
        if((f[dp[i][0]]+f[dp[i][1]]+f[dp[i][2]]+f[dp[i][3]])==n){
            sort(dp[i],dp[i]+4);
            if(  dp[i][0]==dp[ind][0]  &&  dp[i][1]<dp[ind][1]){
                    ind=i;
            }if(dp[i][0]<dp[ind][0]){
                    ind=i;
            }
        }
    }
    for(int i=0;i<4;i++){
        printf("%d%c",dp[ind][i],i==3?'\n':' ');
    }

    return 0;
}