[프로그래머스/Kotlin] 덧칠하기 (Lv. 1)

문제

어느 학교에 페인트가 칠해진 길이가n미터인 벽이 있습니다. 벽에 동아리 · 학회 홍보나 회사 채용 공고 포스터 등을 게시하기 위해 테이프로 붙였다가 철거할 때 떼는 일이 많고 그 과정에서 페인트가 벗겨지곤 합니다. 페인트가 벗겨진 벽이 보기 흉해져 학교는 벽에 페인트를 덧칠하기로 했습니다.

넓은 벽 전체에 페인트를 새로 칠하는 대신, 구역을 나누어 일부만 페인트를 새로 칠 함으로써 예산을 아끼려 합니다. 이를 위해 벽을 1미터 길이의 구역n개로 나누고, 각 구역에 왼쪽부터 순서대로 1번부터 n번까지 번호를 붙였습니다. 그리고 페인트를 다시 칠해야 할 구역들을 정했습니다.

벽에 페인트를 칠하는 롤러의 길이는 m미터이고, 롤러로 벽에 페인트를 한 번 칠하는 규칙은 다음과 같습니다.

• 롤러가 벽에서 벗어나면 안 됩니다.
• 구역의 일부분만 포함되도록 칠하면 안 됩니다.

즉, 롤러의 좌우측 끝을 구역의 경계선 혹은 벽의 좌우측 끝부분에 맞춘 후 롤러를 위아래로 움직이면서 벽을 칠합니다. 현재 페인트를 칠하는 구역들을 완전히 칠한 후 벽에서 롤러를 떼며, 이를 벽을 한 번 칠했다고 정의합니다.

한 구역에 페인트를 여러 번 칠해도 되고 다시 칠해야 할 구역이 아닌 곳에 페인트를 칠해도 되지만 다시 칠하기로 정한 구역은 적어도 한 번 페인트칠을 해야 합니다. 예산을 아끼기 위해 다시 칠할 구역을 정했듯 마찬가지로 롤러로 페인트칠을 하는 횟수를 최소화하려고 합니다.

정수 n, m과 다시 페인트를 칠하기로 정한 구역들의 번호가 담긴 정수 배열 section이 매개변수로 주어질 때 롤러로 페인트칠해야 하는 최소 횟수를 return 하는 solution 함수를 작성해 주세요.

 

제한사항

• 1 ≤ m ≤ n ≤ 100,000
• 1 ≤ section의 길이 ≤ n
  • 1 ≤ section의 원소 ≤ n
  • section의 원소는 페인트를 다시 칠해야 하는 구역의 번호입니다.
  • section에서 같은 원소가 두 번 이상 나타나지 않습니다.
  • section의 원소는 오름차순으로 정렬되어 있습니다.

 

입출력 예

 

풀이방법

1. 주어진 section 배열을 오름차순 정렬한다.

var sortedSec = section.sorted()

 

2. start 변수에 첫번째로 페인트칠 해야 할 구역의 번호를 담고, idx는 1부터 시작한다. 이유는 section의 0번째 인덱스는 무조건 페인트칠 해야하니까 1번째 인덱스부터 비교하면 되기 때문. 그리고 앞서 말한 이유로 0번째 인덱스는 무조건 페인트 칠을 해야하므로 answer를 1 증가 시켰다.

var start = section[0]
var idx = 1
answer++

 

3. while문을 돌며 만약 idx가 section의 사이즈와 같을 때 while문을 종료한다.

생각해보니 언제 while이 멈추는지 알기 때문에 for문으로 해도 될 것 같다.

while(idx != section.size)

 

4. 페인트를 칠해야 할 영역 - 현재 페인트를 칠하기 시작하는 영역이 롤러의 길이보다 크다면 새롭게 페인트를 칠해줘야하므로 start를 현재 페인트를 칠해야 할 영역으로 변경하고 answer를 1 증가시킨다.

그리고 다음 영역으로 가기 위해 idx도 1 증가시킨다.

if(sortedSec[idx] - start >= m){
    start = sortedSec[idx]
    answer++
}
idx++

 

 

주의할 점

1. 코드를 처음 작성할 때 다시 칠하지 않아도 된다면 continue를 했었다. 근데 실행 시간이 오래 걸리면서 코드가 동작하지 않았다. 알고보니 continue를 해주면 바로 다음으로 넘어가기 때문에 idx++ 실행하지 않아서 idx가 계속 고정된 값으로 있었기 때문에 발생한 오류였다. continue의 기능을 다시 한 번 더 상기시켰당.

if(sortedSec[idx] - start < m){
    continue
}else{
    start = sortedSec[idx]
    answer++
}
idx++

 

 

코드

class Solution {
    fun solution(n: Int, m: Int, section: IntArray): Int {
        var answer: Int = 0
        
        var sortedSec = section.sorted()
        var start = section[0]
        var idx = 1
        answer++
        
        while(idx != section.size){
            if(sortedSec[idx] - start >= m){
                start = sortedSec[idx]
                answer++
            }
            
            idx++
        }
        return answer
    }
}

 

 

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

프로그래머스