🚩 题目
有一个 M*N 的矩阵,每个元素的值取自[1,2,3…MN] 的集合,且互不相同。设计一个程序,找出相邻数字的最长路径。
相邻数字的定义:在矩阵中上下左右位置相邻,且值相差为 1 的数字
例如,下图矩阵中相邻数字的最长链接路径为 1-2-3-4
9 | 8 | 7 |
---|---|---|
5 | 6 | 4 |
1 | 2 | 3 |
🎈 解析
dfs。并使用 visited 数组判断访问与否来减少不必要的循环。每次递归结束后,把当前最长的 list 赋值给 maxList。
如果某次得到的 list 长度 > 总数的一半,则它就是最长的路径。直径退出外层循环。在返回前对 list 进行排序。
public class Solution {
ArrayList<Integer> maxList = new ArrayList<Integer>();
ArrayList<Integer> list;
boolean[][] visited;
int xx, yy;
int m, n;
int[] dx = {1,-1,0,0};
int[] dy = {0,0,1,-1};
public List<Integer> calcMaxLen(int[][] grid) {
if (grid == null || grid.length == 0
|| grid[0].length == 0) {
return new ArrayList<>();
}
m = grid.length;
n = grid[0].length;
visited = new boolean[m][n];
out: for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (!visited[i][j]) {
list = new ArrayList<>();
dfs(grid, i, j);
if (list.size() > m*n/2) {
maxList = list;
break out;
}
maxList = maxList.size() > list.size()? maxList: list;
}
}
}
Collections.sort(maxList, new Comparator<Integer>() {
public int compare(Integer o1, Integer o2) {
return o1-o2;
}
});
return maxList;
}
private void dfs(int[][] grid, int i, int j) {
visited[i][j] = true;
list.add(grid[i][j]);
for (int d = 0; d < 4; d++) {
xx = i + dx[d];
yy = j + dy[d];
if (xx >=0 && xx < m && yy >=0 && yy < n
&& !visited[xx][yy] && Math.abs(grid[i][j] - grid[xx][yy]) == 1) {
dfs(grid, xx, yy);
}
}
}
public static void main(String[] args) {
int[][] grid = {
{9,8,7},
{5,6,4},
{1,2,3}};
Solution s = new Solution();
List<Integer> queue = s.calcMaxLen(grid);
System.out.println(queue);
}
}