[BaekJoon] 1977번 : 완전제곱수 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 완전제곱수 입니다.

1977번: 완전제곱수

👨🏻‍💻 문제 풀이

완전제곱수란 1(1의 제곱), 4(2의 제곱), 9(3의 제곱), 16(4의 제곱), 25(5의 제곱)과 같은 수 입니다.

여기서 입력되는 수의 범위가 1 ~ 10,000 인데 10,000은 100의 제곱 수 이므로

def get_perfect_square_numbers():
    return [pow(num, 2) for num in range(1, 101)]

pow를 활용하여 1부터 100 의 완전 제곱수를 먼저 만들어 줍니다.

perfect_square_numbers = get_perfect_square_numbers()

num1 = int(input())
num2 = int(input())

두 개의 수를 입력 받고

square_num = [num for num in perfect_square_numbers if num1 <= num <= num2]

입력 받은 수를 활용하여 아까 만들어둔 숫자에서 두 수 사이의 완전제곱수만 남겨 둡니다.

if len(square_num) == 0:
    answer = [-1]
else:
    answer = sum(square_num), min(square_num)

이 수가 하나도 없을때는 -1을

수가 있을때는 합과 가장 작은 수를 정답으로 하도록 합니다.

for num in answer:
    print(num)

마지막으로 정답을 추출하면 끝!

 

전체 코드는 아래를 참고하시면 됩니다.

👨🏻‍💻 코드 ( Solution )

def get_perfect_square_numbers():
    return [pow(num, 2) for num in range(1, 101)]

def perfect_square(num1, num2, perfect_square_numbers):
    square_num = [num for num in perfect_square_numbers if num1 <= num <= num2]
    
    if len(square_num) == 0:
        answer = [-1]
    else:
        answer = sum(square_num), min(square_num)
    
    return answer
    
    
    
if __name__ == "__main__":
    num1 = int(input())
    num2 = int(input())
    
    perfect_square_numbers = get_perfect_square_numbers()
    answer = perfect_square(num1, num2, perfect_square_numbers)
    
    for num in answer:
        print(num)

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07