백준 알고리즘/구현

컨베이어 벨트 위의 로봇

paysmile 2021. 10. 12. 22:46

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

 

20055번: 컨베이어 벨트 위의 로봇

길이가 N인 컨베이어 벨트가 있고, 길이가 2N인 벨트가 이 컨베이어 벨트를 위아래로 감싸며 돌고 있다. 벨트는 길이 1 간격으로 2N개의 칸으로 나뉘어져 있으며, 각 칸에는 아래 그림과 같이 1부

www.acmicpc.net

#include <iostream>
#include <vector>

using namespace std;

int n, k;
pair<int, int> robot[250]; //exist, value
bool FindZero() {
	int count = 0;
	for (int i = 0; i < 2 * n; i++) {
		if (robot[i].second == 0) count++;
		if (count >= k) {
			return false;
		}
	}
	return true;
}
void MoveBelt() {
	pair<int, int> tmp = robot[0];
	robot[0] = make_pair(0, robot[2 * n - 1].second);
	for (int i = 2 * n - 1; i > 1; i--) {
		robot[i].second = robot[i - 1].second;
		if (i <= n - 1) {
			robot[i].first = robot[i - 1].first;
		}
		else {
			robot[i].first = 0;
		}
	}
	robot[1] = tmp;
}
void MoveRobot() {
	if (robot[n - 1].first == 1) {
		robot[n - 1].first = 0;
		robot[n] = make_pair(0, robot[n].second);
	}
	for (int i = n - 2; i >= 0; i--) {
		if (robot[i].first == 1) {
			if (robot[i + 1].second > 0 && robot[i + 1].first == 0) {
				robot[i].first = 0;
				robot[i + 1] = make_pair(1, robot[i + 1].second - 1);
			}
		}
	}

	if (robot[0].first == 0 && robot[0].second > 0) {
		robot[0].first = 1;
		robot[0].second -= 1;
	}
}
int main(void) {
	cin >> n >> k;
	for (int i = 0; i < 2 * n; i++) {
		int tp;
		cin >> tp;
		robot[i] = make_pair(0, tp);
	}
	int answer = 0;
	while (FindZero()) {
		answer++;
		MoveBelt();
		MoveRobot();
	}

	cout << answer << endl;
}