[解题报告]Minimum Cost to Make at Least One Valid Path in a Grid

题目链接
给定一个有向的M*N矩阵,每个格子均有一个可无消耗行进的方向,若想要向其他三个方向行进则需要花费1个单位时间去反转这个方向
问从左上角到右下角最短消耗多少个单位时间

单源最短路算法解决即可
SPFA(Shortest Path Faster Algorithm)
宽搜过程中针对四个方向目标点当前最短消耗进行松弛
最终输出目的地点最短消耗即可

func minCost(grid [][]int) int {
	lenX := len(grid)
	if lenX == 0 {
		return 0
	}
	lenY := len(grid[0])
	if lenY == 0 {
		return 0
	}
	cost := make([][]int, len(grid))
	for i := range cost {
		cost[i] = make([]int, len(grid[0]))
		for j := range cost[i] {
			cost[i][j] = -1
		}
	}

	cost[0][0] = 0

	type Coordinate struct {
		x, y int
	}

	queue := list.New()
	queue.PushBack(&Coordinate{0, 0})

	dir := [][]int{{0, 1}, {0, -1}, {1, 0}, {-1, 0}}
	for queue.Len() != 0 {
		curCoor := queue.Front().Value.(*Coordinate)

		for i, v := range dir {
			if curCoor.x + v[0] >= 0 &&
				curCoor.x + v[0] < lenX &&
				curCoor.y + v[1] >= 0 &&
				curCoor.y + v[1] < lenY {

				thisCost := 0
				if grid[curCoor.x][curCoor.y] != i + 1 {
					thisCost++
				}

				if cost[curCoor.x + v[0]][curCoor.y + v[1]] == -1 ||
					cost[curCoor.x + v[0]][curCoor.y + v[1]] > cost[curCoor.x][curCoor.y] + thisCost {

					cost[curCoor.x + v[0]][curCoor.y + v[1]] = cost[curCoor.x][curCoor.y] + thisCost

					queue.PushBack(&Coordinate{curCoor.x + v[0], curCoor.y + v[1]})
				}
			}
		}

		queue.Remove(queue.Front())
	}

	return cost[lenX - 1][lenY - 1]
}
发布了15 篇原创文章 · 获赞 2 · 访问量 1346

猜你喜欢

转载自blog.csdn.net/yeehok/article/details/104878159