[파이썬] 프로그래머스 점 찍기

어려운 문제는 아니었는데 시간 초과 때문에 생각보다 시간이 꽤 걸렸다. 

 

def distance(a,b):
    return ((a**2)+(b**2))
 
def solution(k, d):
    answer = 0
 
    # 원 내부에 있나 체크하면 됌
 
    dis = d**2
 
    for i in range(0,d+1, k):      
 
        count = int((dis - i**2) **0.5)
 
        answer += int(count/k+1)
 
 
 
 
 
 
    return answer