校选拔赛B题

//任给一个正整数问能拆成哪四个非负整数的平方和
#include <bits/stdc++.h>
using namespace std;
int biao[3003];//平方数打表
int flag[11000000];//两数平方和标记
int a[11000000];//维护构成平方和的第一个整数
int b[11000000];//维护构成平方和的第二个整数
int main(){
    int n;cin>>n;                   //输入N
    for(int i=0;i<=2237;i++)biao[i]=i*i;//打表，因为2237*2237>5e6已够用
    flag[0]=1;                      //0也是可用的
    for(int i=0;i<=2237;i++){       //遍历
        for(int j=0;j<=2237;j++){   //遍历
            flag[biao[i]+biao[j]]=1;//标记这个数可以是两个非负整数的平方和
            a[biao[i]+biao[j]]=i;   //维护构成平方和的第一个整数
            b[biao[i]+biao[j]]=j;   //维护构成平方和的第二个整数
        }
    }
    for(int i=0;i<5000000;i++){//遍历扫，一定可以扫到的
        if(flag[i]&&flag[n-i]){//I与N-I两个平方和都存在就有结果了
            int x[4];x[0]=a[i],x[1]=b[i],x[2]=a[n-i],x[3]=b[n-i];//得到构成的四个整数
            sort(x,x+4);cout<<x[0]<<' '<<x[1]<<' '<<x[2]<<' '<<x[3]<<endl;//升序输出
            return 0;
        }
    }
    return 0;
}
样例输入：
5
样例输出：
0 0 1 2
 
样例输入：
12
样例输出：
0 2 2 2