백준 17144번 미세먼지 안녕!

문제 링크 : https://www.acmicpc.net/problem/17144

문제 접근

1. T번 동안 먼지를 확산시키고 공기청정기를 작동시키는걸 반복 한 후 남은 먼지의 양을 구해서 출력해주었다.

소스 코드

import java.util.Scanner;
 
public class Main {
    
    static int r, c, t, upY=-1, downY;
    static int map[][];
    static int dy[] = { 0, 0, 1, -1 };
    static int dx[] = { 1, -1, 0, 0 };
    
    public static void cycle() {
        // 위쪽꺼
        for(int i=upY-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<upY; i++) {
            map[i][c-1] = map[i+1][c-1];
        }
        for(int i=c-1; i>1; i--) {
            map[upY][i] = map[upY][i-1];
        }
        // 한칸 채우기
        map[upY][1] = 0;
        
        //아래쪽꺼
        for(int i = downY+1; 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];
        }
        for(int i=r-1; i>downY; i--) {
            map[i][c-1] = map[i-1][c-1];
        }
        for(int i=c-1; i>1; i--) {
            map[downY][i] = map[downY][i-1];
        }
        map[downY][1] = 0;
    }
    
    public static void diffusion() {
        int tmpmap[][] = new int[r][c];
        tmpmap[upY][0] = -1;
        tmpmap[downY][0] = -1;
        
        for(int i=0; i<r; i++) {
            for(int j=0; j<c; j++) {
                if(map[i][j]<=0) continue;
                
                int difcnt = 0;
                int difsu = map[i][j] / 5;
                for(int d=0; d<4; d++) {
                    int ny = i + dy[d];
                    int nx = j + dx[d];
                    
                    if(ny<0 || nx<0 || ny>=r || nx>=c) continue;
                    if(map[ny][nx] == -1) continue;
                    
                    tmpmap[ny][nx] += difsu;
                    difcnt++;
                }
                tmpmap[i][j] += map[i][j] - (difcnt*difsu);
            }
        }
        
        for(int i=0; i<r; i++) {
            for(int j=0; j<c; j++) {
                map[i][j] = tmpmap[i][j];
            }
        }
    }
    
    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];
        
        for(int i=0; i<r; i++) {
            for(int j=0; j<c; j++) {
                int tmpInt = sc.nextInt();
                map[i][j] = tmpInt;
                if(tmpInt == -1) {
                    if(upY == -1) {
                        upY = i;
                    }
                    else {
                        downY = i;
                    }
                }
            }
        }
        
        for(int i=0; i<t; i++) {
            diffusion();
            cycle();
        }
        
        int sum = 0;
        for(int i=0; i<r; i++) {
            for(int j=0; j<c; j++) {
                if(map[i][j]<=0) continue;
                
                sum += map[i][j];
            }
        }
        
        System.out.println(sum);
        
    }
}