1. 程式人生 > 實用技巧 >leetcode[1030. 距離順序排列矩陣單元格]

leetcode[1030. 距離順序排列矩陣單元格]

題目描述

leetcode1030. 距離順序排列矩陣單元格

給出R C 列的矩陣,其中的單元格的整數座標為 (r, c),滿足 0 <= r < R0 <= c < C

另外,我們在該矩陣中給出了一個座標為(r0, c0)的單元格。

返回矩陣中的所有單元格的座標,並按到 (r0, c0) 的距離從最小到最大的順序排,其中,兩單元格(r1, c1)(r2, c2) 之間的距離是曼哈頓距離,|r1 - r2| + |c1 - c2|。(你可以按任何滿足此條件的順序返回答案。

示例 1:

輸入:R = 1, C = 2, r0 = 0, c0 = 0
輸出:[[0,0],[0,1]]
解釋:從 (r0, c0) 到其他單元格的距離為:[0,1]

示例 2:

輸入:R = 2, C = 2, r0 = 0, c0 = 1
輸出:[[0,1],[0,0],[1,1],[1,0]]
解釋:從 (r0, c0) 到其他單元格的距離為:[0,1,1,2]
[[0,1],[1,1],[0,0],[1,0]] 也會被視作正確答案。

示例 3:

輸入:R = 2, C = 3, r0 = 1, c0 = 2
輸出:[[1,2],[0,2],[1,1],[0,1],[1,0],[0,0]]
解釋:從 (r0, c0) 到其他單元格的距離為:[0,1,1,2,2,3]
其他滿足題目要求的答案也會被視為正確,例如 [[1,2],[1,1],[0,2],[1,0],[0,1],[0,0]]。

提示:

  • 1 <= R <= 100
  • 1 <= C <= 100
  • 0 <= r0 < R
  • 0 <= c0 < C

解法1

最容易想到的方法是首先儲存矩陣內所有的點,然後將其按照哈曼頓距離直接排序

func allCellsDistOrder(R int, C int, r0 int, c0 int) [][]int {
	result := make([][]int,0,R*C)
	// 將所有座標存入result中
	for i:=0;i<R;i++{
		for j:=0;j<C;j++{
			result = append(result,[]int{i,j})
		}
	}
    // 定義less函式
	less:=func(i,j int)bool{
        // p,q分別代表result第一層的元素,第一層元素是1維切片,i<j升序,i>j降序
		p,q := result[i],result[j]
		return math.Abs(float64(p[0]-r0)) + math.Abs(float64(p[1]-c0))< math.Abs(float64(q[0]-r0))+math.Abs(float64(q[1]-c0))
	}
    // 呼叫sort.SliceStable()進行排序
	sort.SliceStable(result,less)
	return result
}