题目描述
给你一个下标从 0 开始的二维整数数组 grid ，它的大小为 m x n ，表示一个商店中物品的分布图。数组中的整数含义为：
0 表示无法穿越的一堵墙。1 表示可以自由通过的一个空格子。- 所有其他正整数表示该格子内的一样物品的价格。你可以自由经过这些格子。
从一个格子走到上下左右相邻格子花费 1 步。
同时给你一个整数数组 pricing 和 start ，其中 pricing = [low, high] 且 start = [row, col] ，表示你开始位置为 (row, col) ，同时你只对物品价格在 闭区间 [low, high] 之内的物品感兴趣。同时给你一个整数 k 。
你想知道给定范围 内 且 排名最高 的 k 件物品的 位置 。排名按照优先级从高到低的以下规则制定：
- 距离：定义为从
start 到一件物品的最短路径需要的步数（较近 距离的排名更高）。
- 价格：较低 价格的物品有更高优先级，但只考虑在给定范围之内的价格。
- 行坐标：较小 行坐标的有更高优先级。
- 列坐标：较小 列坐标的有更高优先级。
请你返回给定价格内排名最高的 k 件物品的坐标，将它们按照排名排序后返回。如果给定价格内少于 k 件物品，那么请将它们的坐标 全部 返回。
示例 1：
输入：grid = [[1,2,0,1],[1,3,0,1],[0,2,5,1]], pricing = [2,5], start = [0,0], k = 3
输出：[[0,1],[1,1],[2,1]]
解释：起点为 (0,0) 。
价格范围为 [2,5] ，我们可以选择的物品坐标为 (0,1)，(1,1)，(2,1) 和 (2,2) 。
这些物品的排名为：
- (0,1) 距离为 1
- (1,1) 距离为 2
- (2,1) 距离为 3
- (2,2) 距离为 4
所以，给定价格范围内排名最高的 3 件物品的坐标为 (0,1)，(1,1) 和 (2,1) 。
示例 2：
输入：grid = [[1,2,0,1],[1,3,3,1],[0,2,5,1]], pricing = [2,3], start = [2,3], k = 2
输出：[[2,1],[1,2]]
解释：起点为 (2,3) 。
价格范围为 [2,3] ，我们可以选择的物品坐标为 (0,1)，(1,1)，(1,2) 和 (2,1) 。
这些物品的排名为：
- (2,1) 距离为 2 ，价格为 2
- (1,2) 距离为 2 ，价格为 3
- (1,1) 距离为 3
- (0,1) 距离为 4
所以，给定价格范围内排名最高的 2 件物品的坐标为 (2,1) 和 (1,2) 。
示例 3：
输入：grid = [[1,1,1],[0,0,1],[2,3,4]], pricing = [2,3], start = [0,0], k = 3
输出：[[2,1],[2,0]]
解释：起点为 (0,0) 。
价格范围为 [2,3] ，我们可以选择的物品坐标为 (2,0) 和 (2,1) 。
这些物品的排名为：
- (2,1) 距离为 5
- (2,0) 距离为 6
所以，给定价格范围内排名最高的 2 件物品的坐标为 (2,1) 和 (2,0) 。
注意，k = 3 但给定价格范围内只有 2 件物品。
提示：
m == grid.lengthn == grid[i].length1 <= m, n <= 1051 <= m * n <= 1050 <= grid[i][j] <= 105pricing.length == 22 <= low <= high <= 105start.length == 20 <= row <= m - 10 <= col <= n - 1grid[row][col] > 01 <= k <= m * n
解法
方法一
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 | class Solution:
def highestRankedKItems(
self, grid: List[List[int]], pricing: List[int], start: List[int], k: int
) -> List[List[int]]:
m, n = len(grid), len(grid[0])
row, col, low, high = start + pricing
items = []
if low <= grid[row][col] <= high:
items.append([0, grid[row][col], row, col])
q = deque([(row, col, 0)])
grid[row][col] = 0
while q:
i, j, d = q.popleft()
for a, b in [[0, 1], [0, -1], [1, 0], [-1, 0]]:
x, y = i + a, j + b
if 0 <= x < m and 0 <= y < n and grid[x][y]:
if low <= grid[x][y] <= high:
items.append([d + 1, grid[x][y], x, y])
q.append((x, y, d + 1))
grid[x][y] = 0
items.sort()
return [item[2:] for item in items][:k]
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48 | class Solution {
public List<List<Integer>> highestRankedKItems(
int[][] grid, int[] pricing, int[] start, int k) {
int m = grid.length, n = grid[0].length;
int row = start[0], col = start[1];
int low = pricing[0], high = pricing[1];
List<int[]> items = new ArrayList<>();
if (low <= grid[row][col] && grid[row][col] <= high) {
items.add(new int[] {0, grid[row][col], row, col});
}
grid[row][col] = 0;
Deque<int[]> q = new ArrayDeque<>();
q.offer(new int[] {row, col, 0});
int[] dirs = {-1, 0, 1, 0, -1};
while (!q.isEmpty()) {
int[] p = q.poll();
int i = p[0], j = p[1], d = p[2];
for (int l = 0; l < 4; ++l) {
int x = i + dirs[l], y = j + dirs[l + 1];
if (x >= 0 && x < m && y >= 0 && y < n && grid[x][y] > 0) {
if (low <= grid[x][y] && grid[x][y] <= high) {
items.add(new int[] {d + 1, grid[x][y], x, y});
}
grid[x][y] = 0;
q.offer(new int[] {x, y, d + 1});
}
}
}
items.sort((a, b) -> {
if (a[0] != b[0]) {
return a[0] - b[0];
}
if (a[1] != b[1]) {
return a[1] - b[1];
}
if (a[2] != b[2]) {
return a[2] - b[2];
}
return a[3] - b[3];
});
List<List<Integer>> ans = new ArrayList<>();
for (int i = 0; i < items.size() && i < k; ++i) {
int[] p = items.get(i);
ans.add(Arrays.asList(p[2], p[3]));
}
return ans;
}
}
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34 | class Solution {
public:
vector<vector<int>> highestRankedKItems(vector<vector<int>>& grid, vector<int>& pricing, vector<int>& start, int k) {
int m = grid.size(), n = grid[0].size();
int row = start[0], col = start[1];
int low = pricing[0], high = pricing[1];
vector<tuple<int, int, int, int>> items;
if (low <= grid[row][col] && grid[row][col] <= high)
items.emplace_back(0, grid[row][col], row, col);
queue<tuple<int, int, int>> q;
q.emplace(row, col, 0);
grid[row][col] = 0;
vector<int> dirs = {-1, 0, 1, 0, -1};
while (!q.empty()) {
auto [i, j, d] = q.front();
q.pop();
for (int l = 0; l < 4; ++l) {
int x = i + dirs[l], y = j + dirs[l + 1];
if (x >= 0 && x < m && y >= 0 && y < n && grid[x][y]) {
if (low <= grid[x][y] && grid[x][y] <= high) items.emplace_back(d + 1, grid[x][y], x, y);
grid[x][y] = 0;
q.emplace(x, y, d + 1);
}
}
}
sort(items.begin(), items.end());
vector<vector<int>> ans;
for (int i = 0; i < items.size() && i < k; ++i) {
auto [d, p, x, y] = items[i];
ans.push_back({x, y});
}
return ans;
}
};
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45 | func highestRankedKItems(grid [][]int, pricing []int, start []int, k int) [][]int {
m, n := len(grid), len(grid[0])
row, col := start[0], start[1]
low, high := pricing[0], pricing[1]
var items [][]int
if low <= grid[row][col] && grid[row][col] <= high {
items = append(items, []int{0, grid[row][col], row, col})
}
q := [][]int{{row, col, 0}}
grid[row][col] = 0
dirs := []int{-1, 0, 1, 0, -1}
for len(q) > 0 {
p := q[0]
q = q[1:]
i, j, d := p[0], p[1], p[2]
for l := 0; l < 4; l++ {
x, y := i+dirs[l], j+dirs[l+1]
if x >= 0 && x < m && y >= 0 && y < n && grid[x][y] > 0 {
if low <= grid[x][y] && grid[x][y] <= high {
items = append(items, []int{d + 1, grid[x][y], x, y})
}
grid[x][y] = 0
q = append(q, []int{x, y, d + 1})
}
}
}
sort.Slice(items, func(i, j int) bool {
a, b := items[i], items[j]
if a[0] != b[0] {
return a[0] < b[0]
}
if a[1] != b[1] {
return a[1] < b[1]
}
if a[2] != b[2] {
return a[2] < b[2]
}
return a[3] < b[3]
})
var ans [][]int
for i := 0; i < len(items) && i < k; i++ {
ans = append(ans, items[i][2:])
}
return ans
}
|