#include #include #include #include #include #include using namespace std; class Room { vector> room_; int n_, m_; public: Room(int n, int m) : room_(n, vector(m, 0)), n_(n), m_(m) {} void SetSpace(int r, int c, int num) { room_[r][c] = num; } int GetSpace(int r, int c) const { return room_[r][c]; } }; class Robot { Room& room_; int r_, c_, d_; int n_, m_; int cnt_; public: Robot(Room& room, int r, int c, int d, int m, int n) : room_(room), d_(d), r_(r), c_(c), n_(n), m_(m), cnt_(0){}; void move() { switch (d_) { case 0: r_--; break; case 1: c_++; break; case 2: r_++; break; case 3: c_--; break; default: break; } } int canFront() { // 벽이면 0, 빈칸이면 1, 청소한 곳이면 2 switch (d_) { case 0: { if (r_ == 0) return 0; int what = room_.GetSpace(r_ - 1, c_); if (what == 0) return 1; else if (what == -1) return 2; break; } case 1: { if (c_ == m_ - 1) return 0; int what = room_.GetSpace(r_, c_ + 1); if (what == 0) return 1; else if (what == -1) return 2; break; } case 2: { if (r_ == n_ - 1) return 0; int what = room_.GetSpace(r_ + 1, c_); if (what == 0) return 1; else if (what == -1) return 2; break; } case 3: { if (c_ == 0) return 0; int what = room_.GetSpace(r_, c_ - 1); if (what == 0) return 1; else if (what == -1) return 2; break; } default: break; } return 0; } void backMove() { // 벽이면 exit(0) d_ = (d_ + 2) % 4; if (canFront() == 0) { cout << cnt_; exit(0); } move(); d_ = (d_ + 2) % 4; } void rotate() { d_ = (d_ + 1) % 4; } bool explore() { int dx[4] = {1, -1, 0, 0}; int dy[4] = {0, 0, 1, -1}; for (int i = 0; i < 4; i++) { int cr = r_ + dx[i]; int cc = c_ + dy[i]; if (cr < 0 || cr >= n_ || cc < 0 || cc >= m_) continue; int state = room_.GetSpace(cr, cc); if (state == 0) { return true; } } return false; } void clean() { while (true) { if (room_.GetSpace(r_, c_) == 0) { cnt_++; room_.SetSpace(r_, c_, -1); } if (explore()) { // 주변이 더러움 rotate(); // 90도 회전 if (canFront() == 1) { // 앞으로 갈 수 있으면 move(); // 앞으로 감 } } else { // 주변이 다 깨끗하면 backMove(); // 후진 } } } }; int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); int n, m, r, c, d; cin >> n >> m >> r >> c >> d; Room room(n, m); for (int i = 0; i < n; i++) { for (int j = 0; j < m; j++) { int num; cin >> num; room.SetSpace(i, j, num); } } Robot robot(room, r, c, d, m, n); robot.clean(); return 0; }

제출 #135703

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력: