[프로그래머스] lv.2 점 찍기 python 풀이

풀이

위 예시처럼, x좌표에서 갈 수 있는 모든 y 중 k의 배수가 되는 것만 구하여 total에 더해주는 식으로 풀이

 

문제 

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

코드

def solution(k, d):
    total = 0
    for i in range(0,d+1,k):
        # 이 x좌표에서 될 수 있는 점의 갯수 
        total += (int((d**2 - i**2)**0.5) // k  + 1)
    return total