코딩테스트/프로그래머스

[프로그래머스/PYTHON] 기사단원의 무기

jinsang-2 2024. 5. 29. 16:10

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

from math import sqrt
def solution(number, limit, power):
    answer=0
    for i in range(1,number+1):
        cnt=0
        s=sqrt(i)
        for j in range(1,int(s)+1):
            if i%j==0:
                cnt+=1
        
        if s%1==0:
            cnt=cnt*2-1
        else:
            cnt=cnt*2
        
        if cnt>limit:
            answer+=power
        else:
            answer+=cnt

    return answer