蓝桥杯 大数定理

#include<stdio.h>
#include<math.h>
int f(int n,int a[],int idx){
    if(n==0)    return 1;        //恰好等于 0,则说明恰好等于某几位的平方和 
    if(idx==4)    return 0;        //超过四位数字 返回 0,不符合要求 
    for(int i=(int)sqrt(n);i>=1;i--){
        a[idx]=i;
        if(f(n-i*i,a,idx+1))    //满足要求 
        return 1;                //在这个地方正式退出程序 
    }
    return 0; //没有满足要求的数字和 
}
int main(){
    for(;;){
        int number;
        scanf("%d",&number);
        int a[]={0,0,0,0};        //初始化数组全部为0 
        int r=f(number,a,0);
        printf("%d：%d %d %d %d",r,a[0],a[1],a[2],a[3]);
    }
    return 0;
}