[프로그래머스] 기사단원의 무기 (Python)

문제 링크

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

코드

def solution(number, limit, power):
    arms = []

    for n in range(1, number + 1):
        temp = []
        for i in range(1, int(n ** 0.5)+1):
            if n % i == 0:
                temp.append(i)
                if i ** 2 != n:
                    temp.append(n//i)
        arms.append(power if len(temp) > limit else len(temp))
    
    return sum(arms)