Programmers | 코딩테스트 입문 - 제곱수 판별하기(자바,파이썬)

프로그래머스 Lv.0 제곱수 판별하기

 

문제 설명

어떤 자연수를 제곱했을 때 나오는 정수를 제곱수라고 합니다. 정수 n이 매개변수로 주어질 때, n이 제곱수라면 1을 아니라면 2를 return하도록 solution 함수를 완성해주세요.

---

제한사항

• 1 ≤ n ≤ 1,000,000

 

JAVA

class Solution {
    public int solution(int n) {
        if((int)Math.sqrt(n) == Math.sqrt(n)){
            return 1;
        }
        return 2;
    }
}

제곱근은 Math의 sqrt함수로 구할 수 있다.

 

Python3

import math
def solution(n):
    answer =  1 if n % math.sqrt(n) == 0 else 2
    return answer

파이썬에서도 sqrt를 사용할 수 있다.