[Algorithm] 프로그래머스 Lv.1 : 정수 제곱근 판별

정수 제곱근 판별

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

---

제한 조건

- n은 1이상, 50000000000000 이하인 양의 정수입니다.

---

입출력 예

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

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

n	return
121	144
3	-1

---

문제 풀이

for문을 통해 i를 계속 증가시키는데 그 조건을 for문 괄호 안에 주는 방식이 아닌 for문 안에 if문을 쓰는 방식으로 조건을 제시하도록 한다. i*i가 n보다 클 때에는 -1을 출력시킨다. 이는 양의 정수의 제곱이 아닌 경우 -1을 출력하라는 의미이다. 즉, n이 3일 경우 -1이 정상적으로 리턴된다.

반면, n과 i*i가 같을 경우에는 (i+1)*(i+1)을 출력시킨다. n이 121일 경우, n과 i*i가 같아지게 될 것이고 같아지게 될 때의 i 값은 11(121이 양의 정수 11의 제곱이기 때문에)이다. 최종적으로 (11+1)*(11+1)==144가 정상적으로 리턴된다.

#include <stdio.h>
#include <stdbool.h>
#include <stdlib.h>

long long solution(long long n) {
    for(long long i=1 ; ; i++)
    {
        if(i*i > n)
            return -1;
        else
        {
            if(n == (i*i))
                return (i+1)*(i+1);
        }
    }
}

---

다른 풀이 방식

출처 : 프로그래머스 스쿨

for문을 통해 i*1의 값이 n보다 작거나 같을 때까지 i를 증가시켜준다. 만일 i*i가 n과 같을 때, 즉 양의 정수의 제곱일 경우에는 return (i+1)*(i+1);을 통해 n이 121일 경우 144를 출력시켜줄 수 있으며, 양의 정수의 제곱이 아닌 경우는 -1을 출력시켜주도록 한다. 위의 문제풀이보다 더 간단한 풀이 방식이네요~ ㅎㅎ

long long solution(long long n) {


    for(long long i=1; i*i<=n; i++)
    {
        if(i*i == n)
            return (i+1)*(i+1);
    }
    return -1;
}