[C++] SWEA [모의 SW 역량테스트] 미생물 격리

출처

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

 

 

풀이

#define _CRT_SECURE_NO_WARNINGS 
#include <bits/stdc++.h>
using namespace std;

int T = 0;
int N, M, K;

pair<int, int> get_direction(int d) {
    if (d == 1) { return make_pair(-1, 0); } // 상
    else if (d == 2) { return make_pair(1, 0); } // 하
    else if (d == 3) { return make_pair(0, -1); } // 좌
    else if (d == 4) { return make_pair(0, 1); } // 우
    else { return make_pair(0, 0); }
}

struct germs {
    pair<int, int> location;
    int count;
    int direction;
    germs(int a, int b, int c, int d)
        : location(make_pair(a, b)), count(c), direction(d) {}
};

vector<germs> colonies;

void mergeColonies() {
    map<pair<int, int>, vector<germs>> positionMap;

    // 군집 위치별로 그룹화
    for (germs& g : colonies) {
        if (g.count > 0) {
            positionMap[g.location].push_back(g);
        }
    }

    colonies.clear(); // 기존 군집을 비우고 합친 군집으로 대체

    for (auto& entry : positionMap) {
        vector<germs>& group = entry.second;
        if (group.size() == 1) {
            colonies.push_back(group[0]);
        }
        else {
            // 여러 군집이 있는 경우
            int maxCount = 0;
            germs mergedGerm = group[0];
            mergedGerm.count = 0; // 합산 초기화

            for (germs& g : group) {
                if (g.count > maxCount) {
                    maxCount = g.count;
                    mergedGerm.direction = g.direction; // 방향 업데이트
                }
                mergedGerm.count += g.count;
            }
            colonies.push_back(mergedGerm);
        }
    }
}

void solution(int n, int m, int k) {
    colonies.clear();
    for (int i = 0; i < k; i++) {
        int a, b, c, d;
        cin >> a >> b >> c >> d;
        colonies.push_back(germs(a, b, c, d));
    }

    // M 시간 동안 진행
    for (int i = 0; i < m; i++) {
        map<pair<int, int>, int> newBoard;

        // 군집 이동
        for (germs& g : colonies) {
            if (g.count > 0) {
                g.location.first += get_direction(g.direction).first;
                g.location.second += get_direction(g.direction).second;

                // 경계에 닿았을 때
                if (g.location.first == 0 || g.location.first == n - 1 ||
                    g.location.second == 0 || g.location.second == n - 1) {
                    g.count /= 2;
                    if (g.count > 0) {
                        // 방향 반전
                        if (g.direction == 1) g.direction = 2;
                        else if (g.direction == 2) g.direction = 1;
                        else if (g.direction == 3) g.direction = 4;
                        else if (g.direction == 4) g.direction = 3;
                    }
                }
            }
        }
        mergeColonies();
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> T;
    for (int i = 1; i <= T; i++) {
        cin >> N >> M >> K;
        solution(N, M, K);
        int result = 0;
        for (germs& g : colonies) {
            result += g.count;
        }
        cout << "#" << i << " " << result << endl;
    }
}

 

정말 구현 그자체...이기 때문에...

 

개인적인 구현 꿀팁이 있다면, 정말 말 그대로 컴퓨팅적 사고를 하자!!

 

1. 가장 먼저, 내가 구현해야 할 문제를 정리한다.
2. 문제를 풀기 위해 어떤 절차를 거쳐야 하는지 작성해본다. (분해)
3.  분해한 내용을 바탕으로 함수를 나눠서 정의한다.
4. 나눈 함수를 정의하면서, 필요없는 부분을 삭제하거나 반복되는 게 있는지 생각해보면서 작성한다.
5. 중간중간 검증을 거친다.