[Code Ground][SCPC 1차예선 방속의 거울]

문제

https://www.codeground.org/practice

문제해결

문제를 보고 바로 시뮬레이션 문제라는 것을 직감했다. 방향만 잘 컨트롤 해준다면 문제없이 풀 수 있다. 더 효율적으로도 짤 수 있었겠지만 경우의 수가 별로 없어서 방햘 전환을 if문 만으로 해결하였다.

 

코드

#include <iostream>
#include <stdio.h>
#include <cstring>
using namespace std;
 
int map[1001][1001];
bool visit[1001][1001];
int N,cnt = 0;
pair<int,pair<int,int>> reflect(int y,int x,int dir){
    pair<int,pair<int,int>> p;
    if(map[y][x] == 0){
        if(dir == 1){p.first = y;p.second.first = x+1;p.second.second = dir;return p;}
        else if(dir == 2){p.first = y+1;p.second.first = x;p.second.second = dir;return p;}
        else if(dir == 3){p.first = y;p.second.first = x-1;p.second.second = dir;return p;}
        else if(dir == 4){p.first = y-1;p.second.first = x;p.second.second = dir;return p;}
    }
    else{
        if(!visit[y][x]){cnt++;visit[y][x] = true;}
        if(dir == 1 && map[y][x] == 2){p.first = y+1;p.second.first = x;p.second.second = 2;return p;}
        else if(dir == 1 && map[y][x] == 1){p.first = y-1;p.second.first = x;p.second.second = 4;return p;}
        else if(dir == 2 && map[y][x] == 2){p.first = y;p.second.first = x+1;p.second.second = 1;return p;}
        else if(dir == 2 && map[y][x] == 1){p.first = y;p.second.first = x-1;p.second.second = 3;return p;}
        else if(dir == 3 && map[y][x] == 2){p.first = y-1;p.second.first = x;p.second.second = 4;return p;}
        else if(dir == 3 && map[y][x] == 1){p.first = y+1;p.second.first = x;p.second.second = 2;return p;}
        else if(dir == 4 && map[y][x] == 2){p.first = y;p.second.first = x-1;p.second.second = 3;return p;}
        else{p.first = y;p.second.first = x+1;p.second.second = 1;return p;}
    }
    return p;
}
int main(int argc, char** argv)
{
    int T, test_case;
    cin >> T;
    for(test_case = 0; test_case<T;test_case++)
    {
        cin >> N;
        for(int i = 0;i<N;i++){
            for(int j = 0;j<N;j++){
                scanf("%1d",&map[i][j]);
            }
        }
        int dir = 1,front_x = 0,front_y = 0;
        cnt = 0;
        while(1){
            pair<int,pair<int,int>> p = reflect(front_y,front_x,dir);
            front_x = p.second.first;
            front_y = p.first;
            dir = p.second.second;
            if(front_x>=N || front_y>=N || front_x<0 || front_y < 0){
                break;
            }
        }
        cout << "Case #" << test_case+1 << endl;
        cout << cnt << "\n";
        memset(visit,false,sizeof(visit));
    }
 
    return 0;//Your program should return 0 on normal termination.
}