[BOJ 17144] 미세먼지 안녕!

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

17144번: 미세먼지 안녕!

 

- 문제 요약

 

미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 R×C인 격자판으로 나타냈고, 1×1 크기의 칸으로 나눴다. 구사과는 뛰어난 코딩 실력을 이용해 각 칸 (r, c)에 있는 미세먼지의 양을 실시간으로 모니터링하는 시스템을 개발했다. (r, c)는 r행 c열을 의미한다.

공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.

1초 동안 아래 적힌 일이 순서대로 일어난다.

1. 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.
   • (r, c)에 있는 미세먼지는 인접한 네 방향으로 확산된다.
   • 인접한 방향에 공기청정기가 있거나, 칸이 없으면 그 방향으로는 확산이 일어나지 않는다.
   • 확산되는 양은 Ar,c/5이고 소수점은 버린다.
   • (r, c)에 남은 미세먼지의 양은 Ar,c - (Ar,c/5)×(확산된 방향의 개수) 이다.
2. 공기청정기가 작동한다.
   • 공기청정기에서는 바람이 나온다.
   • 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
   • 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
   • 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.

다음은 확산의 예시이다.

공기청정기의 바람은 다음과 같은 방향으로 순환한다.

방의 정보가 주어졌을 때, T초가 지난 후 구사과의 방에 남아있는 미세먼지의 양을 구해보자.

 

 

- 알고리즘 정리

 

미세먼지가 확산될 때 동시에 확산된다는 점을 생각해서, 확산된 값을 저장하는 배열 add를 만들어서 구현했습니다.

나머지는 모두 문제에서 나온 대로 시뮬레이션을 돌려주면 간단하게 해결할 수 있습니다.

 

 

- 코드 작성

 

#include<bits/stdc++.h>
using namespace std;

#define MAX 51
int r,c,t,arr[MAX][MAX],add[MAX][MAX],dx[4]={0,1,0,-1},dy[4]={-1,0,1,0},r1,r2,result;
bool flag;

void dust(){
	for(int i=0;i<r;i++){
		for(int j=0;j<c;j++){
			int cnt=0;
			int w=arr[i][j]/5;
			if(arr[i][j]==0||arr[i][j]==-1){
				continue;
			}
			for(int k=0;k<4;k++){
				int nx=i+dx[k];
				int ny=j+dy[k];
				if(nx<0||nx>=r||ny<0||ny>=c){
					continue;
				}
				if(arr[nx][ny]==-1){
					continue;
				}
				add[nx][ny]+=w;
				cnt++;
			}
			add[i][j]-=(cnt*w);
		}
	}
	for(int i=0;i<r;i++){
		for(int j=0;j<c;j++){
			arr[i][j]+=add[i][j];
			add[i][j]=0;
		}
	}
}

void clean(){
    result-=arr[r1-1][0];
    result-=arr[r2+1][0];
    for(int i=r1-1;i>0;i--){
    	arr[i][0]=arr[i-1][0];
	}
    for(int i=0;i<c-1;i++){
    	arr[0][i]=arr[0][i+1];
	}
    for(int i=1;i<=r1;i++){
    	arr[i-1][c-1]=arr[i][c-1];
	}
    for(int i=c-1;i>1;i--){
    	arr[r1][i]=arr[r1][i-1];
	}
    arr[r1][1]=0;
    for(int i=r2+1;i<r-1;i++){
    	arr[i][0]=arr[i+1][0];
	}
    for(int i=0;i<c-1;i++){
    	arr[r-1][i]=arr[r-1][i+1];
	}
    for(int i=r-1;i>=r2;i--){
    	arr[i][c-1]=arr[i-1][c-1];
	}
    for(int i=c-1;i>1;i--){
    	arr[r2][i]=arr[r2][i-1];
	}
    arr[r2][1]=0;
}

int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	cin>>r>>c>>t;
	for(int i=0;i<r;i++){
		for(int j=0;j<c;j++){
			cin>>arr[i][j];
			if(arr[i][j]==-1){
				if(!flag){
					r1=i;
					flag=true;
				}
				else{
					r2=i;
				}
			}
			else{
				result+=arr[i][j];
			}
		}
	}
	while(t--){
		dust();
		clean();
	}
	cout<<result;
}

코드 제출 결과