10-dhlee777

[dhlee777]

🔗 문제 링크

토마토

✔️ 소요된 시간

하루

✨ 수도 코드

기존의 2차원배열의 토마토문제는 풀어봤지만 이번에는 3차원배열의 토마토 문제를 풀게되었다. 달라진점은 탐색범위가 기존에는 상,하,좌,우 4군데 였다면 3차원배열에서는 상,하,좌,우,위,아래 총 6군데를 탐색해주어야한다. 이를 위해 좌표정보를 저장할
tomato 구조체를 선언해주었고 토마토의 상태를저장할 tom[][][]배열을 선언해주었다.

입력받는 코드

cin >> col >> row >> height;

	for (int i = 0; i < height; i++) {
		for (int j = 0; j < row; j++) {
			for (int k = 0; k < col; k++) {
				cin >> tom[i][j][k];
				if (tom[i][j][k] == 1)         //익은토마토가 들어있으면
					q.push(tomato(k, j, i));   //탐색시작점을위해 그 좌표를 큐에 넣는다.
			}
		}
	}

tom배열에 상태를 저장할때는 높이,행,열 순으로 입력받는다. 여기서 만약 토마토의 상태가 1이라면 익은 토마토 이므로 탐색 시작점을 위해 그 좌표를 큐에 넣어준다.

bfs

void bfs() {
	while (!q.empty()) {
		tomato t = q.front();
		q.pop();
		for (int i = 0; i < 6; i++) {      //탐색범위(상,하,좌,우,위,아래)
			int _x = t.x1 + x[i];
			int _y = t.y1 + y[i];
			int _z = t.z1 + z[i];
			if (_x < col && _x >= 0 && _y < row && _y >= 0 && _z < height && _z >= 0 && tom[_z][_y][_x] == 0) {
				tom[_z][_y][_x] = tom[t.z1][t.y1][t.x1] + 1;
				q.push(tomato(_x, _y, _z));

			}
		}


	}

}

탐색범위는 상,하,좌,우,위,아래 6군데 이고 탐색조건은 각행과 열의 범위를 만족하고 tom[][][]==0으로 익지 않은 토마토여야 한다. 이때 tom배열은 원래 토마토의 상태를 저장해주는 배열이었으나 bfs에서는 토마토가익는데 걸린 일수를 저장하는 배열로 바뀌게 된다. 탐색에 성공하였다면 tom배열에 그전일수+1 을해준다. 그 후 큐에 좌표정보를 넣어준다.

걸린일수 찾기

int sum = 0;
	for (int i = 0; i < height; i++) {
		for (int j = 0; j < row; j++) {
			for (int k = 0; k < col; k++) {
				if (tom[i][j][k] == 0)     //bfs후 안익은토마토가 있다면 모두익지못하는 상황이므로  -1을 출력
				{
					cout << -1;
					return 0;
				}
				if (sum < tom[i][j][k]) sum = tom[i][j][k]; //bfs 후 걸린시간을 찾기위해 tom배열의 최대값을 찾아 sum에 저장하는과정

			}
		}
	}

	sum == 1 ? cout << 0 : cout << sum - 1;     

bfs를 해준후 안 익은 토마토가 존재한다면 처음부터 모두 익지 못하는 상황이므로 -1을 출력한다. 걸린일수를 찾기위해 sum을 통해 tom배열의 최대값을 찾아준다. 만약 sum이 1이라면 애초에 모두 익어있는상태이므로 0을 출력 아니면 실제 걸린일수인 sum-1을 출력해준다.

📚 새롭게 알게된 내용

기존 2차원토마토 문제와 별 다를게 없었지만 풀어본지 좀 오래돼서 다시 처음부터 생각하느라 시간이 오래걸렸던 거 같다.

[InSange]

구조체로 x,y,z 3개의 변수를 잘 찾은 것 같습니다.
c++에는 tuple이란 기능도 있는 것을 알고 계시나요? python의 tuple과 달리 c++ tuple은 3개의 인자를 가질 수 있습니다. pair는 2개, tuple은 3개

#include <tuple>
queue<tuple<int, int, int>> q;
while (!q.empty())
	{
		int x, y, z;

		x = get<2>(q.front());
		y = get<1>(q.front());
		z = get<0>(q.front());
}

엄청 옛날에 풀었던거라 이런 소소한 재미를 보았던 것 같네요.
추가적으로 day를 탐색할때 저는 bfs에서 비교를 해서 바로 구했던 것 같습니다. 만약 모든 배열을 탐색한다면 똑같겠지만 일부만 탐색한다는 가정하에 모든 배열을 탐색해서 day를 찾는 것보다는 시간을 조금 줄일 수 있지 않을까요?

for (int i = 0; i < 6; i++)
		{
			int nx, ny, nz;
			nx = x + dx[i];
			ny = y + dy[i];
			nz = z + dz[i];

			if (IsOutRange(nx, ny, nz) && tomato[nz][ny][nx] == 0)
			{
				tomato[nz][ny][nx] = tomato[z][y][x] + 1;
				q.push(make_tuple(nz, ny, nx));
				if (tomato[nz][ny][nx] > day)
					day = tomato[nz][ny][nx];
			}
		}

[seongwon030]

푼 지 얼마 안 된 문제라 기억에 남네요. 저도 비슷하게 좌표를 이용해 탐색해 풀었었는데요. 모든 경우를 탐색하다보니 시간복잡도가 크게 나오더라고요. 큐를 이용한 깔끔한 풀이 잘 봤습니다.

[yuyu0830]

배열에 1, 0만 넣을 생각을 했는데 매번 1씩 더해줘서 배열의 최대 값에 따라 날 수를 계산하는 군요! 이런 방식도 있었네요... 그래도 답을 찾기 위해 한번 더 탐색을 해야하는 방식이니 별도 탐색 없이 최대 날짜를 계산할 수 있는 방법이 있으면 찾아보는 것도 좋을 것 같아요!

이제 국룰 코스 하이퍼 틈메이러 풀러 가야겠죠?

[yuyu0830]

이 부분 생김새가 기묘해서 조금 이해하는데 시간이 걸렸네요... 이런 부분은 실행할 함수를 밖으로 빼고 값에 대해서만 삼항연산을 하면 가독성이 좋아질 것 같아요!

cout << sum == 1 ? 0 : sum - 1; 

[dhlee777]

아 함수만 밖으로 뺄수도 있었군요.. 코드가 훨씬깔끔해지는거 같습니다. 잘써먹겠습니다 !

[dhlee777]

구조체로 x,y,z 3개의 변수를 잘 찾은 것 같습니다. c++에는 tuple이란 기능도 있는 것을 알고 계시나요? python의 tuple과 달리 c++ tuple은 3개의 인자를 가질 수 있습니다. pair는 2개, tuple은 3개

#include <tuple>
queue<tuple<int, int, int>> q;
while (!q.empty())
	{
		int x, y, z;

		x = get<2>(q.front());
		y = get<1>(q.front());
		z = get<0>(q.front());
}

엄청 옛날에 풀었던거라 이런 소소한 재미를 보았던 것 같네요. 추가적으로 day를 탐색할때 저는 bfs에서 비교를 해서 바로 구했던 것 같습니다. 만약 모든 배열을 탐색한다면 똑같겠지만 일부만 탐색한다는 가정하에 모든 배열을 탐색해서 day를 찾는 것보다는 시간을 조금 줄일 수 있지 않을까요?

for (int i = 0; i < 6; i++)
		{
			int nx, ny, nz;
			nx = x + dx[i];
			ny = y + dy[i];
			nz = z + dz[i];

			if (IsOutRange(nx, ny, nz) && tomato[nz][ny][nx] == 0)
			{
				tomato[nz][ny][nx] = tomato[z][y][x] + 1;
				q.push(make_tuple(nz, ny, nx));
				if (tomato[nz][ny][nx] > day)
					day = tomato[nz][ny][nx];
			}
		}

헉.. 항상 인자가 3개일때 구조체를 쓸지 pair<int,pair<int,int>>를 쓸지 고민했었는데 tuple 이라는 좋은 도구가 있었군요...! 감사합니다...! bfs에서 바로 비교를 해주는 방식은 생각도 못했습니다...