덧칠하기 - ( 쉽게 생각하자 )

문제 설명

어느 학교에 페인트가 칠해진 길이가 n미터인 벽이 있습니다. 벽에 동아리 · 학회 홍보나 회사 채용 공고 포스터 등을 게시하기 위해 테이프로 붙였다가 철거할 때 떼는 일이 많고 그 과정에서 페인트가 벗겨지곤 합니다. 페인트가 벗겨진 벽이 보기 흉해져 학교는 벽에 페인트를 덧칠하기로 했습니다. 넓은 벽 전체에 페인트를 새로 칠하는 대신, 구역을 나누어 일부만 페인트를 새로 칠 함으로써 예산을 아끼려 합니다. 이를 위해 벽을 1미터 길이의 구역 n개로 나누고, 각 구역에 왼쪽부터 순서대로 1번부터 n번까지 번호를 붙였습니다. 그리고 페인트를 다시 칠해야 할 구역들을 정했습니다. 벽에 페인트를 칠하는 롤러의 길이는 m미터이고, 롤러로 벽에 페인트를 한 번 칠하는 규칙은 다음과 같습니다. 롤러가 벽에서 벗어나면 안 됩니다. 구역의 일부분만 포함되도록 칠하면 안 됩니다. 즉, 롤러의 좌우측 끝을 구역의 경계선 혹은 벽의 좌우측 끝부분에 맞춘 후 롤러를 위아래로 움직이면서 벽을 칠합니다. 현재 페인트를 칠하는 구역들을 완전히 칠한 후 벽에서 롤러를 떼며, 이를 벽을 한 번 칠했다고 정의합니다. 한 구역에 페인트를 여러 번 칠해도 되고 다시 칠해야 할 구역이 아닌 곳에 페인트를 칠해도 되지만 다시 칠하기로 정한 구역은 적어도 한 번 페인트칠을 해야 합니다. 예산을 아끼기 위해 다시 칠할 구역을 정했듯 마찬가지로 롤러로 페인트칠을 하는 횟수를 최소화하려고 합니다. 정수 n, m과 다시 페인트를 칠하기로 정한 구역들의 번호가 담긴 정수 배열 section이 매개변수로 주어질 때 롤러로 페인트칠해야 하는 최소 횟수를 return 하는 solution 함수를 작성해 주세요.

 

풀이

from collections import deque

def erase(m, section):
    erase = 0
    arr = deque(section)
    for i in range(m):
        if arr[i] - arr[0] + 1 <= m and arr[i] != 0: 
            erase += 1  # m칸 조사하고 지울 범위 정하기
    for j in range(erase):
        arr.popleft()
        arr.append(0)
    return arr

def solution(n, m, section):
    arr = deque(section)
    for i in range(abs(len(section)-m)):  #배열 크기 유지
        arr.append(0)
    cnt = 0
    p = 0
    while sum(arr):
        arr = erase(m, arr)
        cnt += 1
        
    return cnt

테스트 데이터가 50개나 있나..;;   12, 35번 데이터에서 시간 초과로 실패했다. 하지만 1,5,19 등 도 3000ms 가 나온 거 보니 좋은 모델은 확실히 아니다.

def erase(m, arr):
    sol = []    # sol = arr 로 썼더니 얕은 복사가 일어났다.
    sol += arr  # arr[0] 을 갱신하면 안되기 때문에 얕은 복사를 방지한다.
    for i in range(m):
        if arr[0]+i in arr:   # arr에서 arr[0] + (0 ~ m) 숫자가 있으면 삭제한다.
            sol.remove(arr[0]+i) 
    return sol

def solution(n, m, section):
    arr = section
    cnt = 0
    while len(arr):
        arr = erase(m, arr)
        cnt += 1
    return cnt

시간이 더 많이 걸렸다. remove 함수가 모든 배열을 조사하느라 시간을 더 쓴 것 같다. 

section 리스트가 항상 증가하는 배열임을 이용해야될 것 같다.

from collections import deque

def split(arr_raw, m):
    arr = deque(arr_raw)
    for i in range(m):     
        if len(arr):       #
            if arr_raw[0] + i in arr:
                arr.popleft()
    return arr
    
def solution(n, m, section):
    answer = 0
    arr = deque(section)
    while len(arr)>0:
        arr = split(arr, m)
        answer += 1
    return answer

이것도 실패. 처음 시도가 더 빠르다

===========================================================================

너무 어렵게 생각했나보다. 처음 페인트를 칠하는 위치를 갱신하여 m을 벗어나면 cnt += 1 을 해주면 됐었다.

def solution(n, m, section):
    answer = 1   
    prev = section[0]    # 칠하는 시작 위치
    for sec in section:
        if sec - prev >= m:  # 위치 차이가 m을 넘어가면
            prev = sec       # 시작 위치 갱신
            answer += 1

    return answer

더 간단한 식도 있었다.  상황을 단순화 시키는 것이 아주 중요하다

def solution(n, m, section):
    n = 0
    k = 0
    for s in section:
        if s > k:        # s가 시작 위치, k를 벗어나야 갱신
            k = s + m - 1  # k는 칠해지는 마지막 영역 
            n += 1       # 카운터 
    return n

 

 

 

출처

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

프로그래머스