LeetCode 每日一题 LCP 13. 寻宝
题目链接：LCP 13.寻宝
题面
我们得到了一副藏宝图，藏宝图显示，在一个迷宫中存在着未被世人发现的宝藏。
迷宫是一个二维矩阵，用一个字符串数组表示。它标识了唯一的入口（用 'S' 表示），和唯一的宝藏地点（用 'T'
表示）。但是，宝藏被一些隐蔽的机关保护了起来。在地图上有若干个机关点（用 'M' 表示）， 只有所有机关均被触发，才可以拿到宝藏。
要保持机关的触发，需要把一个重石放在上面。迷宫中有若干个石堆（用 'O' 表示），每个石堆都有 无限
个足够触发机关的重石。但是由于石头太重，我们一次只能搬 一个 石头到指定地点。
迷宫中同样有一些墙壁（用 '#' 表示），我们不能走入墙壁。剩余的都是可随意通行的点（用 '.'
表示）。石堆、机关、起点和终点（无论是否能拿到宝藏）也是可以通行的。
我们每步可以选择向上/向下/向左/向右移动一格，并且不能移出迷宫。搬起石头和放下石头不算步数。那么，从起点开始，我们最少需要多少步才能最后拿到宝藏呢？如果无法拿到宝藏，返回
-1 。
示例 Sample
示例 1：
输入： ["S#O", "M..", "M.T"]
输出：16
解释：最优路线为： S->O, cost = 4, 去搬石头 O->第二行的M, cost = 3, M机关触发 第二行的M->O, cost = 3,
我们需要继续回去 O 搬石头。 O->第三行的M, cost = 4, 此时所有机关均触发 第三行的M->T, cost = 2，去T点拿宝藏。
总步数为16。
示例 2：
输入： ["S#O", "M.#", "M.T"]
输出：-1
解释：我们无法搬到石头触发机关
示例 3：
输入： ["S#O", "M.T", "M.."]
输出：17
解释：注意终点也是可以通行的。
限制：
1 <= maze.length <= 1001 <= maze[i].length <= 100maze[i].length == maze[j].length- S 和 T 有且只有一个
- 0 <= M的数量 <= 16
- 0 <= O的数量 <= 40，题目保证当迷宫中存在 M 时，一定存在至少一个 O 。
我的题解
这题比较容易超时，需要预处理出所有点（S T O M ,但不含 . #）间的距离。
这里我用 dist[i][j] 表示 i 到 j 的距离（i、j为 S T M），包含需要搬石头的移动距离（T 不需要，其余都要）。
剩下的简单 DP 即可。
typedef pair<int, int> P;
class Solution {
public:
int gao(const vector<string>& maze, P s, P t, map<pair<P, P>, int>&mem) {
const int n = maze.size();
const int m = maze[0].length();
const int dx[] = {1, -1, 0, 0};
const int dy[] = {0, 0, 1, -1};
if(mem.find({s, t}) != mem.end())
return mem[ {s, t}];
vector<vector<bool>>vis(n, vector<bool>(m, false));
vector<vector<int>>dis(n, vector<int>(m, INT_MAX));
queue<P>que;
que.push(s);
vis[s.first][s.second] = true, dis[s.first][s.second] = 0;
while(!que.empty()) {
P u = que.front();
que.pop();
if(u == t) {
mem[ {s, t}] = mem[ {t, s}] = dis[u.first][u.second];
}
for(int i = 0; i < 4; ++i) {
P v(u.first + dx[i], u.second + dy[i]);
if(v.first >= 0 && v.first < n && v.second >= 0 && v.second < m && maze[v.first][v.second] != '#') {
dis[v.first][v.second] = min(dis[v.first][v.second], dis[u.first][u.second] + 1);
if(maze[v.first][v.second] != '.')
mem[ {s, v}] = mem[ {v, s}] = dis[v.first][v.second];
if(!vis[v.first][v.second]) {
vis[v.first][v.second] = true;
que.push(v);
}
}
}
}
if(mem.find({s, t}) != mem.end())
return mem[ {s, t}];
else
return -1;
}
int minimalSteps(vector<string>& maze) {
const int n = maze.size();
const int m = maze[0].length();
P s, t;
vector<P>g(2);
vector<P>h;
map<pair<P, P>, int>mem;
for(int i = 0; i < n; i++) {
for(int j = 0; j < m; j++)
if(maze[i][j] == 'S')
g[0] = {i, j};
else if(maze[i][j] == 'T')
g[1] = {i, j};
else if(maze[i][j] == 'M')
g.push_back({i, j});
else if(maze[i][j] == 'O')
h.push_back({i, j});
}
vector<vector<int>> dis(g.size(), vector<int>(1 << g.size(), INT_MAX));
vector<vector<int>>dist(g.size(), vector<int>(g.size(), INT_MAX));
for(int i = 0; i < (int)g.size(); i++)
for(int j = 0; j < i; j++) {
if(i == 1 || j == 1) {
dist[i][j] = gao(maze, g[i], g[j], mem);
dist[i][j] = dist[j][i] = dist[i][j] == -1 ? INT_MAX : dist[i][j];
} else
for(P k : h) {
int d1 = gao(maze, g[i], k, mem);
if(d1 == -1)
continue;
int d2 = gao(maze, k, g[j], mem);
if(d1 != -1 && d2 != -1 && dist[i][j] > d1 + d2)
dist[i][j] = dist[j][i] = d1 + d2;
}
}
dis[0][1] = 0;/// [addr][status]
for(int i = 1; i < (1 << g.size()); i++) { /// status
for(int j = 0; j < (int)g.size(); j++) {
if(!(i & (1 << j)))
continue;
for(int k = 0; k < (int)g.size(); k++) {
if(!(i & (1 << k)) || k == j)
continue;
if(dist[k][j] == INT_MAX || dis[k][i ^ (1 << j)] == INT_MAX)
continue;
dis[j][i] = min(dis[j][i], dis[k][i ^ (1 << j)] + dist[k][j]);
}
}
}
return dis[1][(1 << g.size()) - 1] == INT_MAX ? -1 : dis[1][(1 << g.size()) - 1];
}
};
