[백준 2828] 사과 담기 게임 (C++)

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

문제

---

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

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

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

입력

---

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

출력

---

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

 

문제 풀이

---

구현 문제.

 

사과가 입력한 순서대로 떨어지고, 바구니가 M칸으로 이루어져 있음에 유의하며 구현하면 된다. 현재 바구니가 차지하고 있는 M개의 칸에 사과가 떨어지고 있다면 움직이지 않아도 된다. 그 공간을 벗어나면 바구니를 움직여야 한다. 사과가 바구니의 왼쪽 끝 칸보다 더 왼쪽에서 떨어지고 있다면 왼쪽으로, 사과가 바구니의 오른쪽 큰 칸보다 더 오른쪽에서 떨어지고 있다면 오른쪽으로 움직이면 된다. 이때 바구니를 움직이는 최솟값은 각각 바구니의 현재 맨 왼쪽 칸 위치 - 사과의 위치, 사과의 위치 - 바구니의 현재 맨 오른쪽 칸 위치가 된다. 해당 칸으로 움직였으므로 바구니의 위치도 재조정해주면 된다.

 

아래는 코드.

#include <iostream>
using namespace std;

int main()
{
	cin.tie(NULL);
	ios::sync_with_stdio(false);

	int N, M, j;
	int applePos;
	cin >> N >> M;
	cin >> j;
	int left = 1;
	int right = M;
	int count = 0;
	for (int i = 0; i < j; i++)
	{
		cin >> applePos;
		if (applePos >= left && applePos <= right)
		{
			continue;
		}
		else if (applePos < left)
		{
			count += left - applePos;
			left = applePos;
			right = left + M - 1;
		}
		else
		{
			count += applePos - right;
			right = applePos;
			left = right - M + 1;
		}
	}
	cout << count << "\n";
	return 0;
}