[프로그래머스 코딩테스트 연습 Lv. 1] 기사단원의 무기

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

프로그래머스

 

def solution(number, limit, power):
    cnt_list = []
    for num in range(1, number+1): # 1부터 number까지 각각 약수 개수 구하기
        cnt = 0
        for n in range(1, int((num+1)**(1/2))+1):
            if num % n == 0 and n**2 != num:
                cnt += 2 # 약수는 항상 짝을 가지므로 짝 개수까지 한 번에 계산
            if n**2 == num:
                cnt += 1
        cnt_list.append(cnt)
        
    for i in range(len(cnt_list)):
        if cnt_list[i] > limit:
            cnt_list[i] = power
                   
    answer = sum(cnt_list)
    
    return answer