2382 미생물 격리

https://www.swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV597vbqAH0DFAVl

실제로 시뮬레이션을 해본다. 매 시간마다 상하좌우에서 현재 위치로 오는 미생물들을 검사하고 다음 상태를 저장한다. 가장자리에서의 예외 처리, 모든 경우의 수 처리, 테스트 케이스마다 초기화에 유의하자.

#include <cstdio>
 
int t,n,m,k,map[100][100][2],tmap[100][100][2],ans;
int dx[4]={1,-1,0,0},dy[4]={0,0,1,-1}; // 하상우좌
 
void solve()
{
    while(m--)
    {
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                int sum=0,t=0,d=0;
                bool f=false;
                for(int k=0;k<4;k++)
                {
                    int nx=i+dx[k],ny=j+dy[k];
                    int &S=map[nx][ny][0],&D=map[nx][ny][1];
                    if(nx<0 || nx>=n || ny<0 || ny>=n || !S)
                        continue;
                    if((D==1 && k==0) || (D==2 && k==1) || (D==3 && k==2) || (D==4 && k==3))
                    {
                        f=true;
                        sum+=S;
                        if(t<S) t=S,d=D;
                    }
                }
 
                if(!f) continue;
                if(i==0 || i==n-1 || j==0 || j==n-1)
                {
                    sum/=2;
                    if(d==1 || d==3) d++;
                    else d--;
                }
                tmap[i][j][0]=sum;
                tmap[i][j][1]=d;
            }
        }
 
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                map[i][j][0]=tmap[i][j][0];
                map[i][j][1]=tmap[i][j][1];
                tmap[i][j][0]=tmap[i][j][1]=0;
            }
        }
    }
 
    ans=0;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<n;j++)
        {
            ans+=map[i][j][0];
            map[i][j][0]=map[i][j][1]=0;
        }
    }
}
 
int main()
{
    scanf("%d",&t);
    for(int T=1;T<=t;T++)
    {
        scanf("%d %d %d",&n,&m,&k);
        while(k--)
        {
            int a,b,c,d;
            scanf("%d %d %d %d",&a,&b,&c,&d);
            map[a][b][0]=c;
            map[a][b][1]=d;
        }
        solve();
        printf("#%d %d\n",T,ans);
    }
    return 0;
}