[C++,구현] 백준 17144번 미세먼지 안녕! 문제풀이

문제링크

백준 17144번 미세먼지 안녕!

17144번: 미세먼지 안녕!

 

이문제 1솔 했다고 생각했는데 결과는 불합이었다...ㅜ 수행시간이 문제였던건가ㅜ 아쉽다. 좀더 실력을 갈고 닦자

지금 다시 풀어보니 백준에서는 통과가 된다.. 차라리 안되면 좋으련만...

 

#include <cstdio>
#include <vector>
using namespace std;
int r,c,t,ans;
int dx[4]={0,1,0,-1};
int dy[4]={1,0,-1,0};
int map[50][50],tmap[50][50],zero[50][50];
vector <int> aircon;
 
void copymap(int (*a)[50],int (*b)[50]){
    for(int i=0;i<r;i++)
        for(int j=0;j<c;j++)
            a[i][j]=b[i][j];
}
 
void spot(int x,int y){
    int dust=map[x][y]/5,can=0;
    if(dust>0){
        for(int i=0;i<4;i++){
            int xx=x+dx[i],yy=y+dy[i];
            if(xx>=0&&xx<r&&yy>=0&&yy<c&&map[xx][yy]!=-1){
                tmap[xx][yy]+=dust;
                can++;
            }
        }
        tmap[x][y]+=map[x][y]-dust*can;
    }
    else if(dust==0)tmap[x][y]+=map[x][y];
 
}
 
void spread(){
    tmap[aircon[0]][0]=-1,tmap[aircon[1]][0]=-1;
    for(int i=0;i<r;i++){
        for(int j=0;j<c;j++){
            if(map[i][j]>0)
            spot(i,j);
        }
    }
    
    copymap(map,tmap);
}
 
void cycling(){
    for(int i=aircon[0]-1;i-1>=0;i--)
        map[i][0]=map[i-1][0];
    for(int i=0;i+1<c;i++)
        map[0][i]=map[0][i+1];
    for(int i=0;i+1<=aircon[0];i++)
        map[i][c-1]=map[i+1][c-1];
    for(int i=c-1;i-1>0;i--)
        map[aircon[0]][i]=map[aircon[0]][i-1];
    map[aircon[0]][1]=0;
    
    
    for(int i=aircon[1]+1;i+1<r;i++)
        map[i][0]=map[i+1][0];
    for(int i=0;i+1<c;i++)
        map[r-1][i]=map[r-1][i+1];
    for(int i=r-1;i-1>=aircon[1];i--)
        map[i][c-1]=map[i-1][c-1];
    for(int i=c-1;i-1>0;i--)
        map[aircon[1]][i]=map[aircon[1]][i-1];
    map[aircon[1]][1]=0;
    
}
 
void count(){
    for(int i=0;i<r;i++){
        for(int j=0;j<c;j++){
            if(map[i][j]!=-1)ans+=map[i][j];
        }
        
    }
    
}
 
void solve(){
    while(t--){
        copymap(tmap,zero);
        spread();
        cycling();
    }
    count();
    printf("%d",ans); 
}
 
main(){
    scanf("%d%d%d",&r,&c,&t);
    for(int i=0;i<r;i++)
        for(int j=0;j<c;j++){
            scanf("%d",&map[i][j]);
            if(map[i][j]==-1)aircon.push_back(i);
        }
        
    solve();
            
    
}

입력을 받고 미세먼지가 퍼지는 spread 함수, 공기청정기가 돌아가는 cycling 함수를

t초 만큼 돌아가게 while문 안에 넣고 마지막에 미세먼지를 count해서 출력하는 알고리즘이다.