题解 | #迷宫问题#

迷宫问题

https://www.nowcoder.com/practice/cf24906056f4488c9ddb132f317e03bc

package main

import (
	"fmt"
)

func main() {
	// m行 n列
	var m, n int
	fmt.Scan(&m, &n)
	vals := make([][]int, m)
	for i := range vals {
		vals[i] = make([]int, n)
		for j := range vals[i] {
			fmt.Scan(&vals[i][j])
		}
	}
	// 最短路径,广搜即可
	queue := []*MyPoint{{
		x: 0, y: 0,
	}}
	// 记忆化
	visited := make(map[struct{ x, y int }]struct{}, 0)
	for len(queue) > 0 {
		p := queue[0]
		queue = queue[1:]
		// 越界判断
		if p.x < 0 || p.x >= m || p.y < 0 || p.y >= n {
			continue
		}
		if _, exist := visited[struct {
			x int
			y int
		}{
			p.x, p.y,
		}]; exist {
			continue
		}
		visited[struct {
			x int
			y int
		}{
			p.x, p.y,
		}] = struct{}{}
		// 墙判断
		if vals[p.x][p.y] == 1 {
			continue
		}
		// 判断是否到达
		if p.x == m-1 && p.y == n-1 {
			// 到达了,从当前结点遍历至初始结点
			link := make([]*MyPoint, 0)
			for p != nil {
				link = append(link, p)
				p = p.parent
			}
			// 逆序打印
			for i := len(link) - 1; i >= 0; i-- {
				fmt.Printf("(%d,%d)\n", link[i].x, link[i].y)
			}
			return
		}
		// 上
		queue = append(queue, &MyPoint{
			x: p.x, y: p.y - 1, parent: p,
		})
		// 下
		queue = append(queue, &MyPoint{
			x: p.x, y: p.y + 1, parent: p,
		})
		// 左
		queue = append(queue, &MyPoint{
			x: p.x - 1, y: p.y, parent: p,
		})
		// 右
		queue = append(queue, &MyPoint{
			x: p.x + 1, y: p.y, parent: p,
		})
	}
}

type MyPoint struct {
	x, y   int
	parent *MyPoint
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务