Pr.2828 사과담기 게임

Pr.2828 사과담기 게임

문제

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

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

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

입력

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

출력

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

예제 입력

5 1
3
1
5
3

예제 출력

6

풀이과정

전제조건

1. 시간 복잡도, 공간 복잡도
   • 시간복잡도 : o(n) 만큼 소요 단순
2. 최대, 최소 조건
   • 바구니 길이 : m
   • 게임 판 : n
3. 사용 알고리즘
   • 단순 구현
4. 반례, 경계값
   • 바구니 경계값 확인

제출 코드

#include <bits/stdc++.h>
using namespace std;
#define y1 aaaaa;


int m, n, j;
int lx, rx, ret, moving;


int main(){
    cin >> n >> m;
    cin >> j;

    lx = 1;
    rx = lx+m-1;

    for(int i = 0; i < j; i++){
        int apple;
        cin >> apple;
        if(apple >= lx && apple <= rx) continue;
        moving = min(abs(apple -lx), abs(apple -rx));
        if(apple > rx){
            lx += moving;
            rx += moving;
        } else if(apple < lx){
            lx -= moving;
            rx -= moving;
        }
        ret += moving;
    }

    cout << ret << endl;
    return 0;
}

정답 코드