Submission #1165421
Source Code Expand
#include "bits/stdc++.h"
using namespace std;
#ifdef _DEBUG
#include "dump.hpp"
#else
#define dump(...)
#endif
//#define int long long
#define rep(i,a,b) for(int i=(a);i<(b);i++)
#define rrep(i,a,b) for(int i=(b)-1;i>=(a);i--)
#define all(c) begin(c),end(c)
const int INF = sizeof(int) == sizeof(long long) ? 0x3f3f3f3f3f3f3f3fLL : 0x3f3f3f3f;
const int MOD = (int)(1e9) + 7;
template<class T> bool chmax(T &a, const T &b) { if (a < b) { a = b; return true; } return false; }
template<class T> bool chmin(T &a, const T &b) { if (b < a) { a = b; return true; } return false; }
int di[4] = { 0,1,0,-1 };
int dj[4] = { 1,0,-1,0 };
char dir[4] = { 'R','D','L','U' };
int H, W, K, sr, sc;
char s[51][51];
int N;
int fr[2510], fc[2510], F[2510], D[2510];
bool inrange(int i, int j) { return i >= 0 && i < H && j >= 0 && j < W; }
void output(string ans) {
if (ans.size() >= 2500) {
cout << ans.substr(0, 2500) << endl;
}
else {
ans += string(2500 - ans.size(), '-');
cout << ans << endl;
}
}
using Weight = int;
using Flow = int;
struct Edge {
int s, d; Weight w; Flow c;
Edge() {};
Edge(int s, int d, Weight w = 1) : s(s), d(d), w(w), c(w) {};
bool operator<(const Edge &e)const { return w < e.w; }
};
using Edges = vector<Edge>;
using Graph = vector<Edges>;
using Array = vector<Weight>;
using Matrix = vector<Array>;
inline ostream &operator<<(ostream &os, const Edge &e) { return (os << '(' << e.s << ", " << e.d << ", " << e.w << ')'); }
void add_edge(Graph &g, int a, int b, Weight w = 1) {
g[a].emplace_back(a, b, w);
g[b].emplace_back(b, a, w);
}
void add_arc(Graph &g, int s, int d, Weight w = 1) { g[s].emplace_back(s, d, w); }
//未探索, 探索中, 探索済
enum { WHITE, GRAY, BLACK };
//単一始点最短経路(負閉路なし)
//Dijkstra O((E+V)logV)
//dist: 始点から各頂点までの最短距離
//戻り値: 最短経路木の親頂点(根は-1)
vector<int> dijkstra(const Graph &g, int s, Array &dist) {
int n = g.size();
vector<int> color(n, WHITE); color[s] = GRAY;
vector<int> prev(n, -1);
dist.assign(n, INF); dist[s] = 0;
using State = tuple<Weight, int, int>; //始点からの最短距離 子 親
priority_queue<State, vector<State>, greater<State>> pq; pq.emplace(0, s, -1);
while (pq.size()) {
Weight d; int v, u; tie(d, v, u) = pq.top(); pq.pop(); //distが最小となるStateを取り出す
if (dist[v] < d)continue; //すでに最短でなければ無視
color[v] = BLACK; prev[v] = u;
for (auto &e : g[v]) {
if (color[e.d] == BLACK)continue;
if (dist[e.d] > dist[v] + e.w) {
dist[e.d] = dist[v] + e.w;
pq.emplace(dist[e.d], e.d, v);
color[e.d] = GRAY;
}
}
}
return prev;
}
auto idx = [&](int i, int j) {return i*W + j; };
//二次元配列からGraphを生成
Graph build(const vector<vector<char>> &v) {
const int H = v.size(), W = v[0].size();
static const int di[4] = { 1,0,-1,0 }, dj[4] = { 0,1,0,-1 };
auto inrange = [&](int i, int j) { return i >= 0 && i < H && j >= 0 && j < W; };
Graph g(H*W);
rep(i, 0, H)rep(j, 0, W) {
if (v[i][j] == '#')continue;
rep(k, 0, 4) {
int ni = i + di[k], nj = j + dj[k];
if (!inrange(ni, nj))continue;
if (v[ni][nj] == '#')continue;
int s = idx(i, j), d = idx(ni, nj);
add_arc(g, s, d);
}
}
return g;
}
struct UnionFind {
vector<int> parent;
int size;
UnionFind(int n) :parent(n, -1), size(n) {}
bool unite(int x, int y) {
x = root(x); y = root(y);
if (x == y)return false;
if (size_of(x) < size_of(y))swap(x, y);
parent[x] += parent[y]; parent[y] = x; size--;
return true;
}
bool same(int x, int y) { return root(x) == root(y); }
int root(int x) { return parent[x] < 0 ? x : parent[x] = root(parent[x]); }
int size_of(int x) { return -parent[root(x)]; }
};
string bfs(int si, int sj, int gi, int gj) {
using P = tuple<int, int, string>;
queue<P> q;
vector<vector<int>> d(H, vector<int>(W, INF));
q.emplace(si, sj, "");
auto inrange = [&](int i, int j) { return i >= 0 && i < H && j >= 0 && j < W; };
while (q.size()) {
int ci, cj; string r; tie(ci, cj, r) = q.front(); q.pop();
if (ci == gi&&cj == gj)return r;
if (!inrange(ci, cj))continue;
if (s[ci][cj] == '#')continue;
if (d[ci][cj] < INF)continue;
d[ci][cj] = r.size();
rep(i, 0, 4) {
q.emplace(ci + di[i], cj + dj[i], r + dir[i]);
}
}
return "";
}
signed main() {
cin.tie(0);
ios::sync_with_stdio(false);
cin >> H >> W >> K >> sr >> sc; sr--, sc--;
vector<vector<char>> v(H, vector<char>(W));
rep(i, 0, H)rep(j, 0, W) {
cin >> s[i][j];
v[i][j] = s[i][j];
}
cin >> N;
rep(i, 0, N) {
cin >> fr[i] >> fc[i] >> F[i] >> D[i];
fr[i]--, fc[i]--;
}
fr[N] = sr, fc[N] = sc;
N++;
Graph g1 = build(v);
Matrix C(N, Array(N));
Graph g2(N);
rep(i, 0, N) {
Array dist;
int s = idx(fr[i], fc[i]);
dijkstra(g1, s, dist);
rep(j, i + 1, N) {
int d = idx(fr[j], fc[j]);
add_edge(g2, i, j, dist[d]);
C[i][j] = C[j][i] = dist[d];
}
}
Graph g3(N);
sort(all(g2[N - 1]));
int a = N - 1, b = g2[N - 1][0].d, c = g2[N - 1][1].d;
add_edge(g3, a, b, C[a][b]);
add_edge(g3, b, c, C[b][c]);
add_edge(g3, c, a, C[c][a]);
//巡回路上にないものの中で巡回路上の点との距離が最小の点を挿入
rep(a, 0, N - 3) {
Edge mini(0, 0, INF);
dump(g3);
rep(i, 0, N) {
if (!g3[i].size()) {
sort(all(g2[i]));
rep(j, 0, N - 1) {
if (g3[g2[i][j].d].size()) {
chmin(mini, g2[i][j]);
break;
}
}
}
}
dump(mini);
int k = mini.s;
Edge ij(0, 0, INF);
rep(i, 0, N) {
if (g3[i].size()) {
rep(j, 0, 2) {
int s = g3[i][j].s, d = g3[i][j].d;
dump(C[s][k], C[d][k], C[s][d]);
chmin(ij, Edge(s, d, C[s][k] + C[d][k] - C[s][d]));
}
}
}
dump(ij);
rep(i, 0, 2) {
if (g3[ij.s][i].d == ij.d) {
g3[ij.s].erase(g3[ij.s].begin() + i);
break;
}
}
rep(i, 0, 2) {
if (g3[ij.d][i].d == ij.s) {
g3[ij.d].erase(g3[ij.d].begin() + i);
break;
}
}
add_edge(g3, ij.s, k, C[ij.s][k]);
add_edge(g3, ij.d, k, C[ij.d][k]);
}
rep(i, 0, N) {
dump(g3[i]);
}
int s = N - 1;
string ans = "";
vector<int> flag(N);
flag[s] = true;
rep(i, 0, N) {
int d1 = g3[s][0].d, d2 = g3[s][1].d;
int d = flag[d1] ? d2 : d1;
flag[d] = true;
dump(s, d);
ans += bfs(fr[s], fc[s], fr[d], fc[d]);
s = d;
}
output(ans);
return 0;
}
Submission Info
Submission Time |
|
Task |
B - Food Collector |
User |
minaminao |
Language |
C++14 (GCC 5.4.1) |
Score |
11786 |
Code Size |
6567 Byte |
Status |
AC |
Exec Time |
5257 ms |
Memory |
20608 KB |
Judge Result
Set Name |
test_01 |
test_02 |
test_03 |
test_04 |
test_05 |
test_06 |
test_07 |
test_08 |
test_09 |
test_10 |
Score / Max Score |
1302 / 20000 |
1219 / 20000 |
1111 / 20000 |
819 / 20000 |
1456 / 20000 |
1297 / 20000 |
1224 / 20000 |
1143 / 20000 |
955 / 20000 |
1260 / 20000 |
Status |
|
|
|
|
|
|
|
|
|
|
Set Name |
Test Cases |
test_01 |
subtask_01_01.txt |
test_02 |
subtask_01_02.txt |
test_03 |
subtask_01_03.txt |
test_04 |
subtask_01_04.txt |
test_05 |
subtask_01_05.txt |
test_06 |
subtask_01_06.txt |
test_07 |
subtask_01_07.txt |
test_08 |
subtask_01_08.txt |
test_09 |
subtask_01_09.txt |
test_10 |
subtask_01_10.txt |
Case Name |
Status |
Exec Time |
Memory |
subtask_01_01.txt |
AC |
1422 ms |
10876 KB |
subtask_01_02.txt |
AC |
5257 ms |
20608 KB |
subtask_01_03.txt |
AC |
2224 ms |
12800 KB |
subtask_01_04.txt |
AC |
254 ms |
3456 KB |
subtask_01_05.txt |
AC |
1525 ms |
10496 KB |
subtask_01_06.txt |
AC |
3559 ms |
18048 KB |
subtask_01_07.txt |
AC |
2310 ms |
13312 KB |
subtask_01_08.txt |
AC |
822 ms |
5504 KB |
subtask_01_09.txt |
AC |
278 ms |
3712 KB |
subtask_01_10.txt |
AC |
2847 ms |
14464 KB |