5653. [모의 SW 역량테스트] 줄기세포배양

swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWXRJ8EKe48DFAUo

SW Expert Academy

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<vector>
#include<queue>
using namespace std;
#define MAXNM 10
#define NMSIZE 705
int dy[] = { 0,1,0,-1 };
int dx[] = { 1,0,-1,0 };
int ret;
int N, M, K, X;//가로, 세로, 시간, 생명력
int board[NMSIZE][NMSIZE];//입력 값
int visit[NMSIZE][NMSIZE];// 큐 배열 체크
struct Data {
	int y, x, num,active,life;
};
queue<Data>q[11];//큐배열
void init() {
	//초기화
	N = M = ret = 0;
	memset(board, 0, sizeof(board));
	memset(visit, 0, sizeof(visit));
	for (int i = 0; i < 11; i++) { while (!q[i].empty())q[i].pop(); }//큐초기화
	//입력
	scanf("%d %d %d", &N, &M, &K);
	for (int i = 350; i < N + 350; i++) {
		for (int j = 350; j < M + 350; j++) {
			scanf("%d", &board[i][j]);
			if (board[i][j] != 0) {
				visit[i][j] = 1;
				q[board[i][j]].push({ i,j,board[i][j],board[i][j],board[i][j] });//종류, 생명력
			}
		}
	}
}
void play() {
	while (K--) {//K기간 동안
		for (int i = 10; i >= 1; i--) {
			int size = q[i].size();
			if (size == 0)continue;//큐 체크
			queue<Data>q1;//큐 저장
			while (size--) {//현재 큐 만큼 돌리기
			
				Data c = q[i].front(); q[i].pop();
				if(c.active!=0)c.active--;
				else {
					Data n;
					for (int dir = 0; dir < 4; dir++) {
						n.y = c.y + dy[dir]; n.x = c.x + dx[dir];
						if (visit[n.y][n.x] == 0 && board[n.y][n.x] == 0) {
							board[n.y][n.x] = c.num;
							n.life = c.num;
							n.active = c.num;
							n.num = c.num;
							visit[n.y][n.x] = c.num;// 배열 체크
							q1.push(n);
						}
					}
					c.life--;
				}
				if (c.life >= 1) {//그 이외의 경우
					q1.push(c);
				}
			}
			while (!q1.empty()) {//재 저장
				q[i].push(q1.front()); q1.pop();
			}
		}

	}
	for (int i = 1; i <= 10; i++) {
		ret += q[i].size();
	}
}
int main(int argc, char** argv)
{
	int test_case;
	int T;
	cin >> T;
	for (test_case = 1; test_case <= T; ++test_case)
	{
		init();
		play();
		printf("#%d %d\n", test_case, ret);
	}
	return 0;//정상종료시 반드시 0을 리턴해야합니다.
}

큐배열을 이용했지만 우선순위 큐를 이용해서 풀이도 가능합니다. 

좀 생각을 해야하는 문제이기 때문에 처음 설계를 잘해야 성공하니 조심하세요