글 읽기 - 어디가 틀렸는지 보이지 않습니다. 예제는 맞습니다.

yuntake94 4년 전 0

송구합니다. 코드가 다소 더럽더라도 이해부탁드립니다.

#include<cstdio>
#include<queue>
#include<vector>
using namespace std;
int m, n, h;
int arr[101][101][101];
int check[101][101][101];
int dx[4] = { 0,0,1,-1 };
int dy[4] = { 1,-1,0,0 };
int ans = 0;
queue<pair<pair<int, int>, int>> v_p;
int Minus_one_One_Check() {
	int flag = false;
	int temp_h = h;
	int h_const = 0;
	while (temp_h--) {
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				if (arr[i][j][h_const] ==0) {
					flag = true;
					break;
				}
			}
			if (flag) {
				break;
			}
		}
		if (flag) {
			break;
		}
	}
	return flag;
}
void TomatoD_Check(int y, int x, int value_h) {
	int flag = false;
	for (int i = 0; i < 4; i++) {
		int mx = dx[i] + x;
		int my = dy[i] + y;
		if (mx >= 0 && mx < m && my >= 0 && my < n) {
			if (arr[my][mx][value_h] == 0 && check[my][mx][value_h] == false) {
				check[my][mx][value_h] = true;
				arr[my][mx][value_h] = 1;
				flag = true;
				v_p.push(make_pair(make_pair(my, mx), value_h));
			}
		}

	}
	if ((value_h + 1) < h && check[y][x][value_h + 1] == false && arr[y][x][value_h + 1] == 0) {
		arr[y][x][value_h + 1] = 1;
		check[y][x][value_h + 1] = true;
		v_p.push(make_pair(make_pair(y, x), value_h + 1));
		flag = true;
	}
	if ((value_h - 1) >= 0 && check[y][x][value_h - 1] == false && arr[y][x][value_h - 1] == 0) {
		arr[y][x][value_h - 1] = 1;
		check[y][x][value_h - 1] = true;
		v_p.push(make_pair(make_pair(y, x), value_h - 1));
		flag = true;
	}
}
void dfs() {
	int v_p_size = v_p.size();
	for (int i = 0; i < v_p_size; i++) {
		int temp_h = v_p.front().second;
		int temp_y = v_p.front().first.first;
		int temp_x = v_p.front().first.second;
		v_p.pop();
		TomatoD_Check(temp_y, temp_x, temp_h);

	}
	ans++;
	if (!Minus_one_One_Check()) {
		printf("%d\n", ans);
		return;
	}
	else
		dfs();



}
int Six_Direction_Check(int y, int x, int temp_h) {
	int flag = false;
	for (int i = 0; i < 4; i++) {
		int mx = dx[i] + x;
		int my = dy[i] + y;
		if (mx >= 0 && mx < m && my >= 0 && my < n) {
			if (arr[my][mx][temp_h] != -1) {
				flag = true;
			}
		}
	}
	if ((temp_h + 1) < h) {
		if (arr[y][x][temp_h + 1] != -1) {
			flag = true;
		}
	}

	if ((temp_h - 1) >= 0) {
		if (arr[y][x][temp_h - 1]) {
			flag = true;
		}
	}
	return flag;


}
int main() {
	scanf("%d %d %d", &m, &n, &h);
	int h_count = 0;
	int temp_h = h;
	while (temp_h--) {
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++)
			{
				scanf("%d", &arr[i][j][h_count]);
	
				if (arr[i][j][h_count] == 1) {
					v_p.push(make_pair(make_pair(i, j), h_count));
				}

			}
		}
		h_count++;
	}
	temp_h = h;
	h_count = 0;
	while (temp_h--) {
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++)
			{
				

				if (arr[i][j][h_count] == 0) {
					int temp_flag = Six_Direction_Check(i, j, h_count);
					if (!temp_flag) {
						printf("-1\n");
						return 0;
					}
				}
			}
		}

		h_count++;
	}
	if (v_p.size() == 0) {
		printf("-1\n");
		return 0;
	}
	int flag = Minus_one_One_Check();
	if (!flag) {
		printf("0\n");
		return 0;
	}

	dfs();





}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

어디가 틀렸는지 보이지 않습니다. 예제는 맞습니다.

yuntake94 4년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말