[코드업] 성실한 개미 문제풀이(C++)

그리디하게 해결하면 됨 / 예외처리 잘 해주기

#include<iostream>
#include<vector>
using namespace std;
/*
<문제>
미로에서 개미의 이동경로 예측하기
막힌 길이 나오면 오른쪽 또는 아래로만 이동
먹이를 찾거나, 더이상 움직일 수 없으면 끝
미로구조는 2차원 배열이고 0은 빈공간, 1은 벽, 2는 먹이
개미는 반드시 (2, 2)에서 출발
 
<입력>
10*10 크기 미로구조 및 먹이위치
 
<출력>
개미의 이동경로를 9로 표시해 출력
 
<풀이>
예외처리 잘 하면서 그리디로 해결
*/
 
int main() {
    vector<vector<int>> miro(10, vector<int>(10, 0));
 
    for (int i = 0; i < 10; i++) {
        for (int j = 0; j < 10; j++) {
            int a; cin >> a; miro[i][j] = a;
        }
    }
    //미로 입력 완료
    
    int a = 1, b = 0;    //개미의 현위치
    while (true) {
        
        if (b + 1 < 10 && miro[a][b + 1] != 1) {    //오른쪽 이동 가능 시
            if (miro[a][++b] == 2) {
                miro[a][b] = 9; break;
            }
            miro[a][b] = 9;
        }
        else if (a + 1 < 10 && miro[a + 1][b] != 1) {    //아래 이동 가능 시
            if (miro[++a][b] == 2) {
                miro[a][b] = 9; break;
            }
            miro[a][b] = 9;
        }
        else {    //둘다 이동 불가능 시
            break;    
        }
    }
 
    for (int i = 0; i < 10; i++) {
        for (int j = 0; j < 10; j++) {
            cout << miro[i][j] << " ";
        }
        cout << endl;
    }
    return 0;
}