#다음과 같이 import를 사용할 수 있습니다.
#import math

def solution(watering_can, N, M):

    NO_WATER = 0
    WATER = 1

    garden = [[NO_WATER] * N for i in range(N)]

    for i in range(M) :
        can = watering_can[i]
        garden[can[0]][can[1]] = WATER
        for j in range(1,can[2]):
            if can[0]+j < N:
                garden[can[0]+j][can[1]] = WATER
            if can[0]-j >= 0:
                garden[can[0]-j][can[1]] = WATER
            if can[1]+j < N:
                garden[can[0]][can[1]+j] = WATER
            if can[1]-j >= 0:
                garden[can[0]][can[1]-j] = WATER

    answer = 0

    for i in range(N):
        for j in range(N):
            if garden[i][j] == NO_WATER:
                answer += 1

    return answer

# 아래는 테스트케이스 출력을 해보기 위한 코드입니다. 아래에는 잘못된 부분이 없으니 위의 코드만 수정하세요.
watering_can1 = [[2,2,2],[0,0,1]]
N1 = 5
M1 = 2
ret1 = solution(watering_can1, N1, M1)
# [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
print("solution 함수의 반환 값은", ret1, "입니다.")

• 클릭하면 바이러스 감염 위험 있음