[코딩테스트] 프로그래머스 덧칠하기 파이썬(Python)

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

프로그래머스

def solution(n, m, section):
    wall = [i+1 for i in range(n)]
    paint = set(wall) - set(section)
    count = 0
    for i in section:
        if i not in paint:
            roller = wall[i-1:i-1+m]
            paint.update(roller)
            count += 1
    return count

저는 집합(set)자료구조를 이용하여 문제를 해결하였습니다.

 

초기에 설정하는 변수는 아래와 같습니다.

• wall : 전체 벽
• paint : 페인트가 칠해져있는 벽
• count : 최소 횟수를 저장할 변수

section을 반복하면서 칠해야하는 벽(i) 을 추출합니다.

 

i벽이 paint에 존재하지 않는다면, 즉 페인트 칠이 되어 있지 않다면
i벽부터 m 길이 만큼의 roller를 생성하고 paint.update()를 하여 페인트 칠을 해줍니다.

 

그리고 count 변수를 1증가 시킵니다.

 

section을 모두 확인했다면 count변수를 반환해줍니다.