Submission #65658841

---

Source Code Expand

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#include <atcoder/all>
using namespace atcoder;

int main(){
    int H, W; cin >> H >> W;
    vector<vector<char>> S(H, vector<char>(W)), ans(H, vector<char>(W, '-'));
    queue<pair<int, int>> q;
    for(int i = 0; i < H; i++){
        for(int j = 0; j < W; j++){
            cin >> S[i][j]; 
            if(S[i][j] == 'E'){
                ans[i][j] = 'E';
                q.push({i, j});
            }  
            if(S[i][j] == '#'){
                ans[i][j] = '#';
            }
        }
    }    
    vector<pair<int, int>> d = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}};
    while(!q.empty()){
        auto [x, y] = q.front(); q.pop();
        for(auto [dx, dy] : d){
            int nx = x + dx, ny = y + dy;
            if(0 <= nx && nx < H && 0 <= ny && ny < W && ans[nx][ny] == '-'){
                if(dx == 1)ans[nx][ny] = '^';
                if(dx == -1)ans[nx][ny] = 'v';
                if(dy == 1)ans[nx][ny] = '<';
                if(dy == -1)ans[nx][ny] = '>';
                q.push({nx, ny});
            }
        }
    }
    for(int i = 0; i < H; i++){
        for(int j = 0; j < W; j++){
            cout << ans[i][j];
        }
        cout << endl;
    }
}

Submission Info

Judge Result