기사단원의 무기

프로그래머스-기사단원의 무기(링크바로가기)

def solution(number, limit, power):
    tmp = []
    for i in range(1,number+1):
        cnt = 0
        for j in range(1,int(i**0.5)+1):
            if(i % j == 0):
                cnt += 2
                if j**2 == i: cnt -=1
            if cnt > limit:
                cnt=power
                break
        tmp.append(cnt)
        
    return sum(tmp)