뮁이의 개발새발

[JAVA] 백준 17144 미세먼지 안녕! (BFS) 본문

카테고리 없음

[JAVA] 백준 17144 미세먼지 안녕! (BFS)

뮁뮁이 2021. 9. 25. 16:12

계속 미세하게 값이 달라서 뭐지 했는데 map의 값이 변경되니까 사방탐색반복문에서 제대로 값이 안나왔었다.. 그래서 사방 반복문 들어가기 전에 먼지를 변수에 저장해주고 그 먼지로 연산하도록 하였다.

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class bj17144 {

	static class node {
		int x;
		int y;
		int dust;

		public node(int x, int y) {
			super();
			this.x = x;
			this.y = y;
		}

		public node(int x, int y, int dust) {
			super();
			this.x = x;
			this.y = y;
			this.dust = dust;
		}

	}

	static int R, C, T;
	static Queue<node> queue;
	static int[][] map;
	static boolean[][] visit;

	// 왼오상하
	static int[] deltax = { -1, 1, 0, 0 };
	static int[] deltay = { 0, 0, -1, 1 };

	public static void main(String[] args) throws IOException {
		BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(in.readLine());
		R = Integer.parseInt(st.nextToken());
		C = Integer.parseInt(st.nextToken());
		T = Integer.parseInt(st.nextToken());

		map = new int[R][C];
		queue = new LinkedList<>();

		int air = 0;
		for (int i = 0; i < R; i++) {
			st = new StringTokenizer(in.readLine());
			for (int j = 0; j < C; j++) {
				map[i][j] = Integer.parseInt(st.nextToken());
				if (map[i][j] == -1) {
					air = i;
				}
			}
		}
		for (int t = 0; t < T; t++) {
			BFS();
			circuit(air);
		}

		System.out.println(sum() + 2);
	}

	private static void BFS() {
		for (int i = 0; i < R; i++) {
			for (int j = 0; j < C; j++) {
				if (map[i][j] > 0) {
					queue.add(new node(j, i, map[i][j]));
				}
			}
		}

		while (!queue.isEmpty()) {
			node temp = queue.poll();
			int cnt = 0;
			int x = temp.x;
			int y = temp.y;
			int dust = temp.dust;
			for (int j = 0; j < 4; j++) {
				int nx = x + deltax[j];
				int ny = y + deltay[j];

				if (nx <= -1 || ny <= -1 || nx >= C || ny >= R) {
					continue;
				}

				if (map[ny][nx] == -1) {
					continue;
				}

				map[ny][nx] += dust / 5;
				cnt++;
			}
			map[y][x] -= (dust / 5) * cnt;
		}
	}

	static void circuit(int air) {

		int newR = air - 1;
		// 윗부분
		for (int i = newR - 1; i > 0; i--) {
			map[i][0] = map[i - 1][0];
		}
		for (int i = 0; i < C - 1; i++) {
			map[0][i] = map[0][i + 1];
		}
		for (int i = 0; i < newR; i++) {
			map[i][C - 1] = map[i + 1][C - 1];
		}
		for (int i = C - 1; i > 0; i--) {
			map[newR][i] = map[newR][i - 1];
		}
		map[newR][1] = 0;

		newR = air;
		// 밑부분
		for (int i = newR; i < R - 1; i++) {
			map[i][0] = map[i + 1][0];
		}
		for (int i = 0; i < C - 1; i++) {
			map[R - 1][i] = map[R - 1][i + 1];
		}
		map[newR][0] = -1;
		for (int i = R - 1; i >= newR; i--) {
			map[i][C - 1] = map[i - 1][C - 1];
		}
		for (int i = C - 1; i > 0; i--) {
			map[newR][i] = map[newR][i - 1];
		}
		map[newR][1] = 0;
	}

	static int sum() {
		int sum = 0;
		for (int i = 0; i < R; i++) {
			for (int j = 0; j < C; j++) {
				sum += map[i][j];
			}
		}
		return sum;
	}
}
Comments