2382. [모의 SW 역량테스트] 미생물 격리

Thanks to 갓무새!!!!

#include <stdio.h>
#include <string.h>
 
typedef struct
{
    int y, x, num, direction;
}bio;
 
int map[101][101][2];
int N, M, K;
bio B[1004];
 
void Go(int *y, int *x, int *num, int *d)
{
    switch (*d)
    {
    case 1://상
        *y -= 1;
        break;
    case 2://하
        *y += 1;
        break;
    case 3://좌
        *x -= 1;
        break;
    case 4://우
        *x += 1;
        break;
    }
 
    if (*y == 0 && *d == 1)
    {
        *num /= 2;
        *d = 2;
    }
    else if (*y == N - 1 && *d == 2)
    {
        *num /= 2;
        *d = 1;
    }
    else if (*x == 0 && *d == 3)
    {
        *num /= 2;
        *d = 4;
    }
    else if (*x == N - 1 && *d == 4)
    {
        *num /= 2;
        *d = 3;
    }
    return;
}
 
 
 
int main()
{
    int Test_Case;
    scanf("%d", &Test_Case);
 
    for (int T = 1; T <= Test_Case; T++)
    {
        memset(map, 0, sizeof(map));
        memset(B, 0, sizeof(B));
        scanf("%d %d %d", &N, &M, &K);
 
        for (int i = 1; i <= K; i++)
        {
            scanf("%d %d %d %d", &B[i].y, &B[i].x, &B[i].num, &B[i].direction);
        }
 
        for (int t = 0; t < M; t++)
        {
            memset(map, 0, sizeof(map));
            for (int i = 1; i <= K; i++)
            {
                if (B[i].num == 0)
                    continue;
 
                Go(&B[i].y, &B[i].x, &B[i].num, &B[i].direction);
 
                //map에 미생물이 없을 경우
                if (map[B[i].y][B[i].x][0] == 0)
                {
                    map[B[i].y][B[i].x][0] = i;
                    map[B[i].y][B[i].x][1] = B[i].num;
                }
                //미생물이 있을 경우
                else if (map[B[i].y][B[i].x][0] != 0)
                {
                    //맵에 있는 미생물의 개수가 현재 보다 클 경우.
                    if (map[B[i].y][B[i].x][1] > B[i].num)
                    {
                        B[map[B[i].y][B[i].x][0]].num += B[i].num;
 
                        B[i].num = B[i].direction = 0;
                    }
                    //맵에 있는 미생물의 개수가 현재 보다 작을 경우.
                    else if (map[B[i].y][B[i].x][1] <  B[i].num)
                    {
                        //기존 맵의 미생물의 num을 임시 저장.
                        int Te = B[map[B[i].y][B[i].x][0]].num;
 
                        //기존 맵의 군집을 reset
                        B[map[B[i].y][B[i].x][0]].num = 0;
                        B[map[B[i].y][B[i].x][0]].direction = 0;
 
                        //맵에 현재 미생물 군집의 정보를 대입
                        map[B[i].y][B[i].x][0] = i;
                        map[B[i].y][B[i].x][1] = B[i].num;
 
                        //기존 군집을 흡수
                        B[i].num += Te;
                    }
                }
            }
        }
 
        int total = 0;
        for (int i = 1; i <= K; i++)
        {
            if (B[i].num)
            {
                total += B[i].num;
            }
        }
 
 
        printf("#%d %d\n", T, total);
    }
 
    return 0;
}