프로그래머스 정수 제곱근 판별

1 일때 생각을 못해서 처음에 한번 틀렸다.

 

#include <string>
#include <vector>

using namespace std;

long long solution(long long n) {
    long long answer = 0;
    for(long long i = 1; i<= n; i++)
    {
        if (n %i == 0 && n/i == i)
            return ((i+1)*(i+1));
    }
    return -1;
}