[Algorithm] Baekjoon 20056번 : 마법사 상어와 파이어볼

https://www.acmicpc.net/problem/20056

20056번: 마법사 상어와 파이어볼

struct Node
{
    int y;
    int x;
    int mass;
    int speed;
    int dir;
};
 
int n, m, k, r, c, mass, dir, speed;
int dir1[8][2] = { {-1,0}, {-1,1}, {0,1}, {1,1}, {1,0}, {1,-1}, {0,-1}, {-1,-1} };
 
vector<Node> arr[52][52];
vector<Node> moveableFireballs;
 
void move()
{
    for (int i = 0; i < moveableFireballs.size(); ++i)
    {
        Node node = moveableFireballs[i];
 
        int nextY = node.y + dir1[node.dir][0] * (node.speed % n);
        int nextX = node.x + dir1[node.dir][1] * (node.speed % n);
        
        if (nextY <= 0 || nextY > n)
        {
            nextY = nextY <= 0 ? n + nextY : nextY - n;
        }
        if (nextX <= 0 || nextX > n)
        {
            nextX = nextX <= 0 ? n + nextX : nextX - n;
        }
 
        node.y = nextY;
        node.x = nextX;
 
        arr[nextY][nextX].push_back(node);
    }
 
    moveableFireballs.clear();
 
    for (int i = 1; i <= n; ++i)
    {
        for (int j = 1; j <= n; ++j)
        {
            if (arr[i][j].size() == 1)
            {
                moveableFireballs.push_back(arr[i][j][0]);
                arr[i][j].clear();
            }
            else if (arr[i][j].size() >= 2) // 파이어볼 나누기.
            {
                int fireballCount = arr[i][j].size();
                int oddCount = 0;
                int evenCount = 0;
                int sumMass = 0;
                int sumSpeed = 0;
 
                for (int k = 0; k < arr[i][j].size(); ++k)
                {
                    if (arr[i][j][k].dir % 2 == 0) ++evenCount;
                    else ++oddCount;
 
                    sumMass += arr[i][j][k].mass;
                    sumSpeed += arr[i][j][k].speed;
                }
 
                if (sumMass / 5 == 0)
                {
                    arr[i][j].clear();
                    continue;
                }
                else
                {
                    arr[i][j].clear();
 
                    int dirStandard = 0;
                    if (oddCount != 0 && evenCount != 0) dirStandard = 1;
 
                    for (int k = 0; k < 4; ++k)
                    {
                        moveableFireballs.push_back({ i,j,sumMass / 5, sumSpeed / fireballCount, dirStandard, });
                        dirStandard += 2;
                    }
                }
            }
        }
    }
}
 
 
int main(void)
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
 
    
    cin >> n >> m >> k;
 
    for (int i = 0; i < m; ++i)
    {
        cin >> r >> c >> mass >> speed >> dir;
 
        moveableFireballs.push_back({ r,c,mass,speed, dir });
    }
 
    for (int i = 0; i < k; ++i)
    {
        move();
        
        int fasd = 0;
    }
    
    int answer = 0;
    for (int i = 0; i < moveableFireballs.size(); ++i)
    {
        answer += moveableFireballs[i].mass;
    }
 
    cout << answer;
    return 0;
}

 

마법사시리즈 두번째문제. 마찬가지로 어렵진 않았는데 엄청 빠르게 풀지는 못했다.

문제설명이 헷갈린것도 있고, 자꾸 자잘한 실수를 한다. ++k를 해야하는데 ++i를 했다던가....

그래도 비슷한 문제가 테스트에 나왔을 때 빠른시간안에 풀기위해 시리즈 전체 다 풀어보긴 할 것 같다.

 

참고로 파이어볼이 4개로 나뉜다는것은 해당단계에서 네 방향으로 흩어진다는게 아니라 그냥 멤버값만 조건에 맞게 가지고 그 자리에 가만~히 있는거다.