[백준] 17144번 미세먼지 안녕!

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

17144번: 미세먼지 안녕!

문제 풀이 :  1초 동안 미세먼지를 확산하고 공기 청정기를 작동시켜야합니다.  spread는 미세먼지를 확산하는 함수입니다. 2차원 배열의 모든 미세먼지를 사방으로 다 퍼뜨립니다. 이때 겹치는 부분이 있으면 미세먼지 수를 더해줘야하기 때문에 map에는 처음 미세먼지 2차원 배열로 해주고 arr 에는 변화되는 미세먼지 배열로 사용합니다. 확산되는 양은 Ar,c/5 이고 남은 미세먼지의 양은 Ar,c - (Ar,c/5)×(확산된 방향의 개수) 를 처리해줍니다. fresh는 공기청정기를 작동시키는 함수입니다. 반 시계방향과 시계방향으로 미세먼지를 이동시키기 위해 for문을 사용합니다. 이때 반시계방향은 ↓←↑→ 방향 순서대로 처리해주어야하고, 시계방향은 ↑←↓→ 방향 순서대로처리해줘야 미세먼지가 제대로 이동합니다. 공기청정기에서 나온 미세먼지는 0으로 해줍니다. (정화) T초가 지난 후 미세먼지 양을 세주면 됩니다.

import java.util.Scanner;
public class Main {
	static int R, C, T, count = 0, X1=0, X2=0, Y1=0, Y2=0,answer = 0;
	static int[][] arr, map, dir = { { -1, 0 }, { 0, -1 }, { 1, 0 }, { 0, 1 } };
	
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		R = sc.nextInt();
		C = sc.nextInt();
		T = sc.nextInt();
		map = new int[R][C];
		arr = new int[R][C];

		for (int i = 0; i < R; i++) {
			for (int j = 0; j < C; j++) {
				map[i][j] = sc.nextInt();
				arr[i][j] = map[i][j];
				if(map[i][j] == -1) {
					if(X1 == 0) X1 = i;//청정기 좌표 저장
					else X2 = i;
				}
			}
		} // end of input

		for(int t=0; t<T; t++) {
			for (int i = 0; i < R; i++) {
				for (int j = 0; j < C; j++) {

					if (map[i][j] != 0 && map[i][j] != -1) {//미세먼지확산
						spread(i, j);
						count = 0;
					}
				}
			}
			
			
			fresh();

			for (int i = 0; i < R; i++) {
				for (int j = 0; j < C; j++) {
					map[i][j] = arr[i][j];
				}
			}
		}//T초
		
			for (int i = 0; i < R; i++) {
				for (int j = 0; j < C; j++) {
					if(arr[i][j] != 0 && arr[i][j] != -1) {
						answer += arr[i][j];
					}
				}
			}
			
			System.out.println(answer);
		
	}



	public static void fresh() {
		// 1. 위의 청정기 가동
		for (int i = X1 - 2; i >= 0; i--) arr[i + 1][Y1] = arr[i][Y1];
		for (int i = 0; i < C -1; i++) arr[0][i] = arr[0][i + 1];
		for (int i = 0; i < X1 ;  i++) arr[i][C-1] =arr[i+1][C-1];
		for(int i = C-2; i>= 1; i--) arr[X1][i+1] = arr[X1][i];
		arr[X1][Y1+1] = 0;//청정기에서 나오면 미세먼지 없어짐
		//2. 아래 청정기 가동
		for (int i = X2 + 1; i < R-1; i++) arr[i][Y2] = arr[i+1][Y2];
		for (int i = Y2 + 1; i < C; i++ )arr[R-1][i-1] = arr[R-1][i];
		for (int i = R-2; i >=X2; i--)arr[i+1][C-1] = arr[i][C-1];
		for (int i = C-2; i> 0; i--) arr[X2][i+1] = arr[X2][i];
		arr[X2][Y2+1] = 0;
	}

	private static void spread(int x, int y) {
		for (int i = 0; i < dir.length; i++) {
			int nx = x + dir[i][0];
			int ny = y + dir[i][1];

			if (isInside(nx, ny) && map[nx][ny] != -1) {
				arr[nx][ny] = arr[nx][ny] + map[x][y] / 5;
				count++;
			}
		}
		arr[x][y] = arr[x][y] - (map[x][y] / 5) * count;

	}

	public static boolean isInside(int x, int y) {
		return x >= 0 && x < R && y >= 0 && y < C;
	}
}