最新消息:XAMPP默认安装之后是很不安全的,我们只需要点击左方菜单的 "安全"选项,按照向导操作即可完成安全设置。

[LeetCode] 675. Cut Off Trees for Golf Event

XAMPP下载 admin 690浏览 0评论
 Problem
You are asked to cut off trees in a forest for a golf event. The forest is represented as a non-negative 2D map, in this map:

0 represents the obstacle can’t be reached.
1 represents the ground can be walked through.
The place with number bigger than 1 represents a tree can be walked through, and this positive number represents the tree’s height.
You are asked to cut off all the trees in this forest in the order of tree’s height – always cut off the tree with lowest height first. And after cutting, the original place has the tree will become a grass (value 1).

You will start from the point (0, 0) and you should output the minimum steps you need to walk to cut off all the trees. If you can’t cut off all the trees, output -1 in that situation.

You are guaranteed that no two trees have the same height and there is at least one tree needs to be cut off.

Example 1:
Input:
[
[1,2,3],
[0,0,4],
[7,6,5]
]
Output: 6
Example 2:
Input:
[
[1,2,3],
[0,0,0],
[7,6,5]
]
Output: -1
Example 3:
Input:
[
[2,3,4],
[0,0,5],
[8,7,6]
]
Output: 6
Explanation: You started from the point (0,0) and you can cut off the tree in (0,0) directly without walking.
Hint: size of the given matrix will not exceed 50×50.

Solution
class Solution {
int[][] dirs = new int[][]{{-1,0},{1,0},{0,-1},{0,1}};
public int cutOffTree(List<List<Integer>> forest) {
//create a list of int[] {value, x, y}
List<int[]> trees = new ArrayList<>();
for (int i = 0; i < forest.size(); i++) {
for (int j = 0; j < forest.get(0).size(); j++) {
int value = forest.get(i).get(j);
if (value > 1) trees.add(new int[]{i, j, value});
}
}

Collections.sort(trees, (a, b)->(a[2]-b[2]));
int res = 0, x = 0, y = 0;
for (int[] tree: trees) {
int dist = bfs(forest, x, y, tree[0], tree[1]);
if (dist < 0) return -1;
else {
res += dist;
x = tree[0];
y = tree[1];
}
}
return res;
}
private int bfs(List<List<Integer>> forest, int x, int y, int tx, int ty) {
int m = forest.size(), n = forest.get(0).size();
Queue<int[]> queue = new LinkedList<>();
queue.add(new int[]{x, y});
boolean[][] visited = new boolean[m][n];
visited[x][y] = true;
int dist = 0;
while (!queue.isEmpty()) {
int size = queue.size();
for (int j = 0; j < size; j++) {
int[] cur = queue.poll();
if (cur[0] == tx && cur[1] == ty) return dist;
for (int i = 0; i < 4; i++) {
int nx = cur[0]+dirs[i][0];
int ny = cur[1]+dirs[i][1];
if (nx < 0 || nx >= m || ny < 0 || ny >= n ||
visited[nx][ny] || forest.get(nx).get(ny) <= 0) continue;
visited[nx][ny] = true;
queue.add(new int[]{nx, ny});
}
}
dist++;
}
return -1;
}
}

转载请注明:XAMPP中文组官网 » [LeetCode] 675. Cut Off Trees for Golf Event

您必须 登录 才能发表评论!