望着时间滴答滴答的流过,我不曾改变过 . . .

牛客2018校招 1. 拼多多 迷宫寻路

题目描述
假设一个探险家被困在了地底的迷宫之中,要从当前位置开始找到一条通往迷宫出口的路径。迷宫可以用一个二维矩阵组成,有的部分是墙,有的部分是路。迷宫之中有的路上还有门,每扇门都在迷宫的某个地方有与之匹配的钥匙,只有先拿到钥匙才能打开门。请设计一个算法,帮助探险家找到脱困的最短路径。如前所述,迷宫是通过一个二维矩阵表示的,每个元素的值的含义如下 0-墙,1-路,2-探险家的起始位置,3-迷宫的出口,大写字母-门,小写字母-对应大写字母所代表的门的钥匙

输入描述:
迷宫的地图,用二维矩阵表示。第一行是表示矩阵的行数和列数M和N
后面的M行是矩阵的数据,每一行对应与矩阵的一行(中间没有空格)。M和N都不超过100, 门不超过10扇。
输出描述:
路径的长度,是一个整数

示例1
输入
5 5
02111
01a0A
01003
01001
01111
输出
7

看到这道题目,求脱困的最短路径,首先就想到了bfs, 但是这里有一点不同的是(难受的是):
1.他有10扇门,10把钥匙,怎么去拿钥匙去开门,钥匙和门怎么对应?
2.同样一个点,拿到钥匙a,b,c的时候,和只拿到钥匙b。这样的时候明显是不同的。
因为 : 这个时候,如果遇到了门A,拿到钥匙a,b,c的时候,可以打开,但是只拿到钥匙b的时候,不能打开。

看评论,一个巧妙地做法是这样的。
利用一个map[x][y][key] , key 用来记录钥匙的状态,横坐标为x,纵坐标为y,
钥匙的状态 就用二进制数表示 最多10 把钥匙 那就是1024
比如我现在有第二把钥匙和第四把钥匙 那么我的钥匙状态就是 0101000000 也就是 320

代码如下:

import java.util.*;

public class Main{
    
    static class Node{
        int x;
        int y;
        int key;
        int step;
        public Node(int x, int y, int key, int step){
            this.x = x;
            this.y = y;
            this.key = key;
            this.step = step;
        }
    }
    
    public static void main(String[] args){
        Scanner in = new Scanner(System.in);
        int N = in.nextInt();
        int M = in.nextInt();
        in.nextLine();
        char[][] G = new char[N][M];
        for (int i = 0;i< N;i++)
            G[i] = in.nextLine().toCharArray();
        for (int i = 0; i< N;i++){
            for (int j=0;j<M;j++){
                if (G[i][j] == '2'){
                    System.out.println(bfs(i,j, N, M, G));
                    return;
                }
            }
        }
    }
    
    private static int bfs(int si, int sj, int N, int M, char[][] G){
        Queue<Node> queue = new LinkedList<>();
        int[][][] mp = new int[101][101][1025];
        int[][] next = {{-1,0},{0,-1},{1,0},{0,1}};
        
        queue.offer(new Node(si,sj, 0, 0));
        while (!queue.isEmpty()){
            Node node = queue.poll();
            for (int i = 0; i < 4 ; i++){
                int x = node.x + next[i][0];
                int y = node.y + next[i][1];
                int key = node.key;
                if(x < 0 || x >= N || y < 0 || y >= M || G[x][y] == '0')    continue;
                else if (G[x][y] == '3')    return node.step +1;
                else if (G[x][y] <= 'z' && G[x][y] >= 'a')    key = key | (1 << (G[x][y] - 'a'));
                else if (G[x][y] <= 'Z' && G[x][y] >= 'A' && (key & (1 << G[x][y] - 'A') ) == 0)    continue;
                if (mp[x][y][key] == 0){
                    mp[x][y][key] = 1;
                    queue.offer(new Node(x,y,key,node.step + 1));
                }
            }
        }
        return -1;
    }
    
    
}
posted @ 2019-04-12 09:25  whyaza  阅读(422)  评论(0编辑  收藏  举报