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

https://school.programmers.co.kr/learn/courses/30/lessons/12934#

프로그래머스

---

아이디어

1. 소인수분해를 해서 인수들을 배열에 저장한다.

2. 인수들 중 하나라도 홀수개라면 제곱수가 아니므로, -1을 리턴

3. 어떤 양수 x를 찾고 (x+1)의 제곱을 리턴한다.

 

제출 코드

def solution(n):

    factor = [] # 소인수들을 리스트에 저장한다.
    i = 2
    while i <= n:
        if n % i == 0:
            factor.append(i)
            n /= i
        else:
            i += 1
    
    factor_set = set(factor)
    
    # 원래의 수 x 구하기!!
    x = 1
    for i in factor_set:
        if factor.count(i) % 2 != 0: # 개수가 짝수개가 아니면 제곱수가 아니다
            return -1
        x *= pow(i,factor.count(i)//2)
    return (x+1) ** 2

 

다른 사람들 풀이

다들.. sqrt 썼네.... 힝