Submission #486781

#TimeUsernameProblemLanguageResultExecution timeMemory
486781davi_bartPatkice II (COCI21_patkice2)C++14
110 / 110
294 ms74832 KiB
#pragma GCC optimize("O3")
#include <bits/stdc++.h>
using namespace std;
#define ll long long
// #define int ll
#define fi first
#define se second
#define ld long double
#define pb push_back
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
char v[2010][2010];
bool vis[2010][2010];
int dd[2010][2010];
int prec[2010][2010];
int N, M;
void solve(int start) {
    deque<tuple<int, int, int> > q;
    q.push_front({0, start, -1});
    while (!q.empty()) {
        auto [d, pos, last] = q.front();
        q.pop_front();
        int x = pos / 10000, y = pos % 10000;
        if (vis[x][y]) continue;
        vis[x][y] = 1;
        dd[x][y] = d;
        prec[x][y] = last;
        if (v[x][y] == 'x') return;

        if (x) {
            (v[x][y] != '^') ? q.push_back({d + 1, (x - 1) * 10000 + y, pos}) : q.push_front({d, (x - 1) * 10000 + y, pos});
        }
        if (x < N - 1) {
            (v[x][y] != 'v') ? q.push_back({d + 1, (x + 1) * 10000 + y, pos}) : q.push_front({d, (x + 1) * 10000 + y, pos});
        }
        if (y) {
            (v[x][y] != '<') ? q.push_back({d + 1, x * 10000 + y - 1, pos}) : q.push_front({d, x * 10000 + y - 1, pos});
        }
        if (y < M - 1) {
            (v[x][y] != '>') ? q.push_back({d + 1, x * 10000 + y + 1, pos}) : q.push_front({d, x * 10000 + y + 1, pos});
        }
    }
}
signed main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin >> N >> M;
    pair<int, int> start, fine;
    for (int i = 0; i < N; i++) {
        string a;
        cin >> a;
        for (int j = 0; j < M; j++) {
            v[i][j] = a[j];
            if (a[j] == 'o') start = {i, j};
            if (a[j] == 'x') fine = {i, j};
        }
    }
    solve(start.fi * 10000 + start.se);
    // for (int i = 0; i < N; i++) {
    //     for (int j = 0; j < M; j++) {
    //         dd[i][j]--;
    //         cout << dd[i][j] << " ";
    //     }
    //     cout << endl;
    // }
    pair<int, int> pos = fine;
    while (1) {
        int k = prec[pos.fi][pos.se];
        if (k == start.fi * 10000 + start.se) break;
        int x = k / 10000, y = k % 10000;
        if (x < pos.fi) v[x][y] = 'v';
        if (x > pos.fi) v[x][y] = '^';
        if (y < pos.se) v[x][y] = '>';
        if (y > pos.se) v[x][y] = '<';
        pos = {x, y};
    }

    cout << dd[fine.fi][fine.se] - 1 << '\n';
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < M; j++) {
            cout << v[i][j];
        }
        cout << '\n';
    }
}

Compilation message (stderr)

Main.cpp: In function 'void solve(int)':
Main.cpp:20:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17'
   20 |         auto [d, pos, last] = q.front();
      |              ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...