본문 바로가기

Solved.ac/Class4

[Java] 백준 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초가 지난 후 구사과의 방에 남아있는 미세먼지의 양을 구해보자.

[입력]

첫째 줄에 R, C, T (6 ≤ R, C ≤ 50, 1 ≤ T ≤ 1,000) 가 주어진다.

둘째 줄부터 R개의 줄에 Ar,c (-1 ≤ Ar,c ≤ 1,000)가 주어진다. 공기청정기가 설치된 곳은 Ar,c가 -1이고, 나머지 값은 미세먼지의 양이다. -1은 2번 위아래로 붙어져 있고, 가장 윗 행, 아랫 행과 두 칸이상 떨어져 있다.

[출력]

첫째 줄에 T초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.

[예제 입력]

(더 많은 입력예제는 사이트에서...)

[풀이 과정]

구현, 시물레이션 분류에 속하는 문제이다. 문제에서 원하는 대로 구현을 하면된다.

 

문제를 크게 나눠보자면 이렇다.

 

1 )  1초마다 미세먼지가 네 방향으로 확산된다

2 )  그 후 공기 청정기가 작동된다

3 )  T 초 만큼 1) -> 2) 과정을 반복한다.

4 )  남은 미세먼지 총 합을  구한다.

 

 

[1단계 : 미세먼지를 확산시키자]

 

먼저 미세먼지가 어떻게 확산이 되는지 알아야한다.

위의 예시를 가지고 설명하자면 미세먼지가 30인 칸이 확산이 될때 (좌 우 하) 칸으로 30/5만큼씩 확산이 된다. 

그리고 그  칸은 30 - (30/5)*3 = 12가 된다.

30 확산

그 다음 7의 위치였던 칸에서 확산된다.  이때 주의할점은  7+6(30에서 나눠진) =13 에서 5로 나누면 안된다. 값이 변경되기전의 7에서 나누어야한다. 따라서 (좌 하) 로 7/5 만큼 더해지고 그 칸은 13  - (7/5)*2 =11이된다.

7 확산

총 과정을 보자면 이러하다!

 

미세먼지가 확산되어 값이 변경되기전의 값을 2차원배열(new_array)을 만든후 값을 복사하여 변경 전의 미세먼지 값을 가지고 연산할 수 있도록 하였다.

    public static void spreadDust(){//미세먼지 확산

        copyArray(); 
        //new_array = 확산되기전 미세먼지 값을 가지는 2차원 배열
        //array = 확산되서 값이 변경된 2차원 배열
        /*
         public static void copyArray(){

        for(int i=0; i<R ;i++){
            for(int j=0; j<C; j++){
                new_array[i][j]=array[i][j];
            }
        }
    }
    */

        for(int i = 0 ; i<R; i++) {
             for (int j = 0 ; j<C;j++) {
              if(array[i][j]>0){ //미세먼지 칸일때 
                  if(array[i][j]/5 <0)continue; //확산될 값이 0이면 넘어간다.
                   int spread = new_array[i][j]/5; //상하좌우로 확산되어지는 값
                   int count = 0; //확산된 칸의 수

                  for (int k = 0; k < 4; k++) {
                      int next_X = i + dot_X[k];
                      int next_Y = j + dot_Y[k];
					  //2차원배열 밖으로 벗어날 수 없다.
                      if (next_X < 0 || next_X >= R || next_Y < 0 || next_Y >= C) continue;
                      if (array[next_X][next_Y] == -1) continue; //공기청정기 칸으로 갈 수 없다.

                      array[next_X][next_Y] += spread; //확산된 값만큼 더해준다.
                      count++;
                  }
                  array[i][j] -= spread * count; //확산된 칸 수  만큼 빼준다.
              }

    }
}
    }

 

 

[2단계 : 공기청정기 작동]

 

 

공기청정기의 행(R) 좌표를 알아내어 사진과 번호와 방향대로 반복문을 통해 값을 1칸씩 당겨주었다. 코드를 보면 이해가 갈 것이다. 그리고 공기청정기 바로 오른쪽의 칸으로 바람이 나오기 때문에 0으로 초기화 시켜준다. 4방향으로 값을 당겨오는 코드 구현이 이 문제에서 조금 까다로웠던거 같다.

 

 public static void cleanDust(){

		//cleaner_dot = 공기청정기 좌표(위)
        
        int up = cleaner_dot;
        int down =cleaner_dot+1;
        
        
        //반시계방향
        for(int i= up -1; i >0;i--){ //1
            array[i][0]=array[i-1][0];
        }
        for(int i=0;i<C-1;i++){      //2
            array[0][i]=array[0][i+1];
        }
        for(int i =0;i<up;i++){      //3
            array[i][C-1]=array[i+1][C-1];
        }   
        for(int i = C-1; i>1;i--){    //4
            array[up][i]=array[up][i-1];
        }
        array[up][1]=0; //공기청정기에서 나오는 바람

        //시계방향
        for(int i= down+1;i<R-1;i++){  //1
            array[i][0]= array[i+1][0];
        }

        for(int i = 0;  i<C-1;i++){    //2
            array[R-1][i] =array[R-1][i+1];
        }

        for(int i= R-1; i>down;i--){    //3
            array[i][C-1]=array[i-1][C-1];
        }
        for(int i=C-1; i>1;i--){        //4
            array[down][i]=array[down][i-1];
        }
        array[down][1]=0;// 공기청정기에서 나오는 바람

    }

 

[3단계 : T초 후 미세먼지 의 총 합을 구한다]

 

1초마다 1,2단계가 진행되니 T초라면 1,2단계를  T번 반복하면된다. 그 후 미세먼지 총 합을 구하면 정답이 나온다!

    public static int totalDust(){

        int sum =0;
        for(int i = 0; i<R ; i++){
            for(int j = 0 ; j <C;j++){
                if(array[i][j]!=-1){ //공기청정기 자리 값은 더해주면안된다.
                    sum+=array[i][j];
                }
            }

        }
        return sum;
    }

 

[코드]

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

public class Main {
    static int R,C,T;
    static int [][]array,new_array;
    static int [] dot_X = {0,0,1,-1};
    static int [] dot_Y = {-1,1,0,0};
    static int cleaner_dot =-1;
    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine()," ");
        R = Integer.parseInt(st.nextToken()); //행
        C = Integer.parseInt(st.nextToken()); //열
        T = Integer.parseInt(st.nextToken()); //초
        array =  new int[R][C];
        new_array = new int[R][C];


        //2차원 배열 초기화
        for(int i = 0 ; i < R ; i++){
            st = new StringTokenizer(br.readLine(), " ");
            for(int j = 0 ; j< C;j++){
                array[i][j] = Integer.parseInt(st.nextToken());
                if(cleaner_dot==-1 && array[i][j]==-1){
                   cleaner_dot=i; //공기청정기 (위) 위치
                }
            }
        }

        while ( T -->0){
            spreadDust();
            cleanDust();
        }
        System.out.print(totalDust());

    }
    public static void copyArray(){

        for(int i=0; i<R ;i++){
            for(int j=0; j<C; j++){
                new_array[i][j]=array[i][j];
            }
        }
    }

    public static void spreadDust(){//미세먼지 확산

        copyArray();

        for(int i = 0 ; i<R; i++) {
             for (int j = 0 ; j<C;j++) {
              if(array[i][j]>0){
                  if(array[i][j]/5 <0)continue;
                   int spread = new_array[i][j]/5;
                   int count = 0;

                  for (int k = 0; k < 4; k++) {
                      int next_X = i + dot_X[k];
                      int next_Y = j + dot_Y[k];

                      if (next_X < 0 || next_X >= R || next_Y < 0 || next_Y >= C) continue;
                      if (array[next_X][next_Y] == -1) continue;

                      array[next_X][next_Y] += spread;
                      count++;
                  }
                  array[i][j] -= spread * count;
              }

    }
}
    }

    public static void cleanDust(){ //공기청정기 작동


        int up = cleaner_dot;
        int down =cleaner_dot+1;
        //반시계
        for(int i= up -1; i >0;i--){
            array[i][0]=array[i-1][0];
        }
        for(int i=0;i<C-1;i++){
            array[0][i]=array[0][i+1];
        }
        for(int i =0;i<up;i++){
            array[i][C-1]=array[i+1][C-1];
        }
        for(int i = C-1; i>1;i--){
            array[up][i]=array[up][i-1];
        }
        array[up][1]=0; //공기청정기에서 나오는 바람

        //시계
        for(int i= down+1;i<R-1;i++){
            array[i][0]= array[i+1][0];
        }

        for(int i = 0;  i<C-1;i++){
            array[R-1][i] =array[R-1][i+1];
        }

        for(int i= R-1; i>down;i--){
            array[i][C-1]=array[i-1][C-1];
        }
        for(int i=C-1; i>1;i--){
            array[down][i]=array[down][i-1];
        }
        array[down][1]=0;// 공기청정기에서 나오는 바람

    }


    public static int totalDust(){ //미세먼지 총 합

        int sum =0;
        for(int i = 0; i<R ; i++){
            for(int j = 0 ; j <C;j++){
                if(array[i][j]!=-1){
                    sum+=array[i][j];
                }
            }

        }
        return sum;
    }

}

 

[결과]