leetcode **LCP 13. 寻宝(待研究)

【题目】**LCP 13. 寻宝

我们得到了一副藏宝图,藏宝图显示,在一个迷宫中存在着未被世人发现的宝藏。

迷宫是一个二维矩阵,用一个字符串数组表示。它标识了唯一的入口(用 ‘S’ 表示),和唯一的宝藏地点(用 ‘T’ 表示)。但是,宝藏被一些隐蔽的机关保护了起来。在地图上有若干个机关点(用 ‘M’ 表示),只有所有机关均被触发,才可以拿到宝藏。

要保持机关的触发,需要把一个重石放在上面。迷宫中有若干个石堆(用 ‘O’ 表示),每个石堆都有无限个足够触发机关的重石。但是由于石头太重,我们一次只能搬一个石头到指定地点。

迷宫中同样有一些墙壁(用 ‘#’ 表示),我们不能走入墙壁。剩余的都是可随意通行的点(用 ‘.’ 表示)。石堆、机关、起点和终点(无论是否能拿到宝藏)也是可以通行的。

我们每步可以选择向上/向下/向左/向右移动一格,并且不能移出迷宫。搬起石头和放下石头不算步数。那么,从起点开始,我们最少需要多少步才能最后拿到宝藏呢?如果无法拿到宝藏,返回 -1 。

示例 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 <= 100
1 <= maze[i].length <= 100
maze[i].length == maze[j].length
S 和 T 有且只有一个
0 <= M的数量 <= 16
0 <= O的数量 <= 40,题目保证当迷宫中存在 M 时,一定存在至少一个 O 。

【解题思路1】状态压缩动态规划

旅行商问题 分支限界法 记忆化搜索

class Solution {
    
    
    int[] dx = {
    
    1, -1, 0, 0};
    int[] dy = {
    
    0, 0, 1, -1};
    int n, m;

    public int minimalSteps(String[] maze) {
    
    
        n = maze.length;
        m = maze[0].length();
        // 机关 & 石头
        List<int[]> buttons = new ArrayList<int[]>();
        List<int[]> stones = new ArrayList<int[]>();
        // 起点 & 终点
        int sx = -1, sy = -1, tx = -1, ty = -1;
        for (int i = 0; i < n; i++) {
    
    
            for (int j = 0; j < m; j++) {
    
    
                if (maze[i].charAt(j) == 'M') {
    
    
                    buttons.add(new int[]{
    
    i, j});
                }
                if (maze[i].charAt(j) == 'O') {
    
    
                    stones.add(new int[]{
    
    i, j});
                }
                if (maze[i].charAt(j) == 'S') {
    
    
                    sx = i;
                    sy = j;
                }
                if (maze[i].charAt(j) == 'T') {
    
    
                    tx = i;
                    ty = j;
                }
            }
        }
        int nb = buttons.size();
        int ns = stones.size();
        int[][] startDist = bfs(sx, sy, maze);

        // 边界情况:没有机关
        if (nb == 0) {
    
    
            return startDist[tx][ty];
        }
        // 从某个机关到其他机关 / 起点与终点的最短距离。
        int[][] dist = new int[nb][nb + 2];
        for (int i = 0; i < nb; i++) {
    
    
            Arrays.fill(dist[i], -1);
        }
        // 中间结果
        int[][][] dd = new int[nb][][];
        for (int i = 0; i < nb; i++) {
    
    
            int[][] d = bfs(buttons.get(i)[0], buttons.get(i)[1], maze);
            dd[i] = d;
            // 从某个点到终点不需要拿石头
            dist[i][nb + 1] = d[tx][ty];
        }

        for (int i = 0; i < nb; i++) {
    
    
            int tmp = -1;
            for (int k = 0; k < ns; k++) {
    
    
                int midX = stones.get(k)[0], midY = stones.get(k)[1];
                if (dd[i][midX][midY] != -1 && startDist[midX][midY] != -1) {
    
    
                    if (tmp == -1 || tmp > dd[i][midX][midY] + startDist[midX][midY]) {
    
    
                        tmp = dd[i][midX][midY] + startDist[midX][midY];
                    }
                }
            }
            dist[i][nb] = tmp;
            for (int j = i + 1; j < nb; j++) {
    
    
                int mn = -1;
                for (int k = 0; k < ns; k++) {
    
    
                    int midX = stones.get(k)[0], midY = stones.get(k)[1];
                    if (dd[i][midX][midY] != -1 && dd[j][midX][midY] != -1) {
    
    
                        if (mn == -1 || mn > dd[i][midX][midY] + dd[j][midX][midY]) {
    
    
                            mn = dd[i][midX][midY] + dd[j][midX][midY];
                        }
                    }
                }
                dist[i][j] = mn;
                dist[j][i] = mn;
            }
        }

        // 无法达成的情形
        for (int i = 0; i < nb; i++) {
    
    
            if (dist[i][nb] == -1 || dist[i][nb + 1] == -1) {
    
    
                return -1;
            }
        }
        
        // dp 数组, -1 代表没有遍历到
        int[][] dp = new int[1 << nb][nb];
        for (int i = 0; i < 1 << nb; i++) {
    
    
            Arrays.fill(dp[i], -1);
        }
        for (int i = 0; i < nb; i++) {
    
    
            dp[1 << i][i] = dist[i][nb];
        }
        
        // 由于更新的状态都比未更新的大,所以直接从小到大遍历即可
        for (int mask = 1; mask < (1 << nb); mask++) {
    
    
            for (int i = 0; i < nb; i++) {
    
    
                // 当前 dp 是合法的
                if ((mask & (1 << i)) != 0) {
    
    
                    for (int j = 0; j < nb; j++) {
    
    
                        // j 不在 mask 里
                        if ((mask & (1 << j)) == 0) {
    
    
                            int next = mask | (1 << j);
                            if (dp[next][j] == -1 || dp[next][j] > dp[mask][i] + dist[i][j]) {
    
    
                                dp[next][j] = dp[mask][i] + dist[i][j];
                            }
                        }
                    }
                }
            }
        }

        int ret = -1;
        int finalMask = (1 << nb) - 1;
        for (int i = 0; i < nb; i++) {
    
    
            if (ret == -1 || ret > dp[finalMask][i] + dist[i][nb + 1]) {
    
    
                ret = dp[finalMask][i] + dist[i][nb + 1];
            }
        }

        return ret;
    }

    public int[][] bfs(int x, int y, String[] maze) {
    
    
        int[][] ret = new int[n][m];
        for (int i = 0; i < n; i++) {
    
    
            Arrays.fill(ret[i], -1);
        }
        ret[x][y] = 0;
        Queue<int[]> queue = new LinkedList<int[]>();
        queue.offer(new int[]{
    
    x, y});
        while (!queue.isEmpty()) {
    
    
            int[] p = queue.poll();
            int curx = p[0], cury = p[1];
            for (int k = 0; k < 4; k++) {
    
    
                int nx = curx + dx[k], ny = cury + dy[k];
                if (inBound(nx, ny) && maze[nx].charAt(ny) != '#' && ret[nx][ny] == -1) {
    
    
                    ret[nx][ny] = ret[curx][cury] + 1;
                    queue.offer(new int[]{
    
    nx, ny});
                }
            }
        }
        return ret;
    }

    public boolean inBound(int x, int y) {
    
    
        return x >= 0 && x < n && y >= 0 && y < m;
    }
}

猜你喜欢

转载自blog.csdn.net/XunCiy/article/details/107653901
今日推荐