# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
977034 | farica | Mecho (IOI09_mecho) | C++14 | 160 ms | 7260 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#define int long long
using namespace std;
const int MAX_N = 4e5 + 5;
bool visited[800][800];
void solve() {
int n, s, sx, sy;
cin >> n >> s;
vector<vector<int>>dist(n, vector<int>(n, -1));
string grid[n];
queue<pair<int,int>>q;
for(int i=0; i<n; ++i) {
cin >> grid[i];
for(int j=0; j<n; ++j) {
if(grid[i][j] == 'H') {
q.push({i, j});
dist[i][j] = 0;
} else if(grid[i][j] == 'M') sx=i, sy=j;
}
}
while(!q.empty()) {
int i = q.front().first, j = q.front().second;
q.pop();
if(i && dist[i-1][j] == -1 && grid[i-1][j] != 'T') {
dist[i-1][j] = dist[i][j] + 1;
q.push({i-1, j});
}
if(j && dist[i][j-1] == -1 && grid[i][j-1] != 'T') {
dist[i][j-1] = dist[i][j] + 1;
q.push({i, j-1});
}
if(i<n-1 && dist[i+1][j] == -1 && grid[i+1][j] != 'T') {
dist[i+1][j] = dist[i][j] + 1;
q.push({i+1, j});
}
if(j<n-1 && dist[i][j+1] == -1 && grid[i][j+1] != 'T') {
dist[i][j+1] = dist[i][j] + 1;
q.push({i, j+1});
}
}
priority_queue<pair<int, pair<int,int>>>pq;
pq.push({dist[sx][sy], {sx, sy}});
while(!pq.empty()) {
int val = pq.top().first, x = pq.top().second.first, y = pq.top().second.second, eD=abs(x-sx) + abs(y-sy);
pq.pop();
if(visited[x][y]) continue;
visited[x][y] = 1;
if(dist[x][y] < eD/s + (eD % s > 0 ? 1 : 0)) continue;
if(grid[x][y] == 'D') {
cout << val - (eD/s + (eD % s > 0 ? 1 : 0)) << endl;
return;
}
if(x && grid[x-1][y] != 'T') pq.push({min(val, dist[x-1][y]), {x-1, y}});
if(y && grid[x][y-1] != 'T') pq.push({min(val, dist[x][y-1]), {x, y-1}});
if(x+1<n && grid[x+1][y] != 'T') pq.push({min(val, dist[x+1][y]), {x+1, y}});
if(y+1<n && grid[x][y+1] != 'T') pq.push({min(val, dist[x][y+1]), {x, y+1}});
}
cout << -1 << endl;
}
signed main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int t = 1;
while(t--) solve();
return 0;
}
Compilation message (stderr)
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |