[每日一题] 四平方和

#include <iostream>
#include <cmath>

using namespace std;

int cal_square_sum(int i, int j, int k, int l) {
    return i*i + j*j + k*k + l*l;
}

int main() {
	int n = 0;
    cin >> n;
	
	int upper1 = sqrt(n/4) + 1;
	int upper2 = sqrt(n/3) + 1;
    int upper3 = sqrt(n/2) + 1;
    int upper4 = sqrt(n) + 1;
    
    for (int i = 0 ; i <= upper1 ; ++i) {
    		    
		for (int j = i ; j <= upper2 ; ++j) {
        	if (i*i + j*j >= n) break;
			for (int k = j ; k <= upper3 ; ++k) {
                if (i*i + j*j + k*k >= n) break;
                for (int l = k ; l <= upper4 ; ++l) {
                    if (i*i + j*j + k*k + l*l > n) break;
                    if (cal_square_sum(i, j, k, l) == n) {
						cout << i << " " << j << " " << k << " " << l;
                        return 0;
                    }
                }
            }
        }
    }

    return 0;
}