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

문제 설명
임의의 양의 정수 n에 대해, n이 어떤 양의 정수 x의 제곱인지 아닌지 판단하려 합니다.
n이 양의 정수 x의 제곱이라면 x+1의 제곱을 리턴하고, n이 양의 정수 x의 제곱이 아니라면 -1을 리턴하는 함수를 완성하세요.

제한 사항
n은 1이상, 50000000000000 이하인 양의 정수입니다.

 

입출력 예
n return
121 144
3 -1

 

입출력 예 설명
입출력 예#1
121은 양의 정수 11의 제곱이므로, (11+1)를 제곱한 144를 리턴합니다.

입출력 예#2
3은 양의 정수의 제곱이 아니므로, -1을 리턴합니다.

---

#include <string>
#include <vector>

using namespace std;

long long solution(long long n)
{
    long long answer = 0;
    long long x = 0;
    
    for(int i = 1; i * i <= n ; ++i)
    {
        if(n == i * i) // n이 양의 정수 x(i)의 제곱 이라면
        {
            x = i;
            answer = (x + 1) * (x + 1);
        }
        else
        {
            answer = -1;
        }
    }
    return answer;
}

함수를 사용하지 않고 풀어보려고 for과 if를 사용해서 풀었는데 테스트 18개중 5개가 시간 초과가 떴다.

그래서 원인이 제한 사항인 n은 50000000000000 이하인 양의 정수여서 for 문에서 시간이 오래 걸리는줄 알았다.

하지만 다른 사람의 풀이에서 어떤 분이 c++에서 정수형끼리의 곱셈을 했을 때 나오는 값도 정수형이 된다고 해서 i를 long long으로 수정했다. (i가 int일때 i * i인 n도 int)

#include <string>
#include <vector>

using namespace std;

long long solution(long long n)
{
    long long answer = 0;
    long long x = 0;
    
    for(long long i = 1; i * i <= n ; ++i) // long long i로 변경
    {
        if(n == i * i)
        {
            x = i;
            answer = (x + 1) * (x + 1);
        }
        else
        {
            answer = -1;
        }
    }
    return answer;
}

테스트가 모두 통과가 되었다.