Определить процедуру, позволяющую распознавать полные квадраты

#include <iostream>
#include <cmath>
 
using namespace std;
 
int square(int n){
    
    if (n == 1)
        return 1;
    
    int flag = 0;
    for (int i = 2; i < n;)
        if (n % i == 0){
            flag = 1;
            if ((n / i) % i == 0)
                n /= (i * i);
            else
                return 0;
        }else
            ++i;
    
    if (flag == 1)
        return 1;
    else
        return 0;       
}
 
int main(){
   
   int n;
   cout << "n = ";cin >> n;
   
   for (int i = 0; i < n; ++i){
    int a;
        cout << "×èñëî: "; cin >> a;
        for (int k = 1; k <= a / 2; ++k)
            if (square(a - k * k) == 1){
                cout << "-------------------" << endl;
                cout << a << " = " << k << "^2 + " 
                      << (int)sqrt(a - k * k) << "^2" << endl;              
                break;
            }
    }
}