백준 사과 담기 게임 코틀린

https://www.acmicpc.net/problem/2828

2828번: 사과 담기 게임

오늘은 백준에서 사과담기 게임 문제를 풀어봤습니다! 

 

문제

상근이는 오락실에서 바구니를 옮기는 오래된 게임을 한다. 스크린은 N칸으로 나누어져 있다. 스크린의 아래쪽에는 M칸을 차지하는 바구니가 있다. (M<N) 플레이어는 게임을 하는 중에 바구니를 왼쪽이나 오른쪽으로 이동할 수 있다. 하지만, 바구니는 스크린의 경계를 넘어가면 안 된다. 가장 처음에 바구니는 왼쪽 M칸을 차지하고 있다.

스크린의 위에서 사과 여러 개가 떨어진다. 각 사과는 N칸중 한 칸의 상단에서 떨어지기 시작하며, 스크린의 바닥에 닿을때까지 직선으로 떨어진다. 한 사과가 바닥에 닿는 즉시, 다른 사과가 떨어지기 시작한다.

바구니가 사과가 떨어지는 칸을 차지하고 있다면, 바구니는 그 사과가 바닥에 닿을 때, 사과를 담을 수 있다. 상근이는 사과를 모두 담으려고 한다. 이때, 바구니의 이동 거리의 최솟값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 M이 주어진다. (1 ≤ M < N ≤ 10) 둘째 줄에 떨어지는 사과의 개수 J가 주어진다. (1 ≤ J ≤ 20) 다음 J개 줄에는 사과가 떨어지는 위치가 순서대로 주어진다.

출력

모든 사과를 담기 위해서 바구니가 이동해야 하는 거리의 최솟값을 출력한다.

예제 입력 1 

5 1

3

1

5

3

예제 출력 1 

6

 

---

저는 각 바구니의 머리와 꼬리의 위치에 정보를 가지고 있다가 

각각에 사과 위치가 머리와 꼬리 위치와 비교해서

오른쪽으로 이동하는지 왼쪽으로 이동하는지 확인하는 과정을 거칩니다!

 

예를 들어서

바구니의 길이가 2라고 가정할때 머리의 위치는 1 꼬리 위치는 2라고 할 수 있습니다!

사과가 3의 위치에서 떨어질 때 3은 꼬리보다 크기 때문에 3-2 =1 만큼 오른쪽으로 이동한다는 것을 알 수 있습니다

그래서 이동 후 머리의 위치는 2 꼬리의 위치는 3이 됩니다

 

반대로 1에서 떨어질때 머리의 위치(2) 보다 작기 때문에 

2-1 =1 만큼 왼쪽으로 이동 해야 하기 때문에 

 

머리의 위치는 1 꼬리는 2 가 됩니다 

 

그러면서 3-2=1 과 2-1=1 의 값들을 더하고 println 로 출력하면 됩니다!

 

 

fun main(){
    var answer = 0
    var str = readLine()
    var count = readLine()?.toInt()
    var list = ArrayList<Int>()
    for(i in 0 until count!!){
        var num = readLine()?.toInt()
        if (num != null) {
            list.add(num)
        }
    }

    var chars = str!!.split(" ")

    var bucketWidth = chars[1].toInt()
    var head = 1
    var tail = bucketWidth


    for(i in list.indices){
        var apple = list.get(i)
        if(tail<apple){

            var temp = apple-tail

            answer += temp
            head +=temp
            tail+=temp

        }else if(head>apple){

            var temp=head -apple

            answer += temp
            head -=temp
            tail-=temp
        }
    }

println(answer)
}