如何解决使用dfs和Map进行二叉树的垂直顺序遍历
我正在尝试对二叉树进行垂直顺序遍历,这个问题是:https://leetcode.com/problems/vertical-order-traversal-of-a-binary-tree/
我的方法:我先对树中每个节点的深度进行深度搜索,其中根为0。左侧的任何内容为当前根-1,右侧的任何内容为当前根root +1。我还有一个HashMap,其中的键是节点的值(位置匹配的键),值是所有具有相同位置的TreeNode(键值)的链表。 / p>
当我打印出HashMap的值时,我看到它具有正确的元素分组。
问题是当我返回最终的List时,整数列表的顺序不正确,应该是从最大的负数到最大的正数-而是打印出与根位置相同的元素第一。我认为这是因为我不能使用负数作为我的HashMap的键-但是我找不到解决办法。
例如:
如果输入树是:
程序应返回:[[9],[3,15],[20],[7]]
但是,我的返回:[[3,15],[9],[20],[7]]
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val,TreeNode left,TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<List<Integer>> verticalTraversal(TreeNode root) {
/*dfs getting value of node & adding it to list*/
List<List<Integer>> list = new ArrayList<>();
Map<Integer,List<Integer>> map = new HashMap<>();
if(root == null) {
return list;
}
dfs(root,map,0);
for(Map.Entry<Integer,List<Integer>> m : map.entrySet()) {
List<Integer> temp = new ArrayList<>();
for(int i : m.getValue()) {
temp.add(i);
}
list.add(temp);
}
return list;
}
public static void dfs(TreeNode root,Map<Integer,List<Integer>> map,int curVal) {
if(root == null) {
return;
}
// System.out.println("root val = " + root.val);
// System.out.println("curVal = " + curVal);
if(map.containsKey(curVal)) {
List temp = map.get(curVal);
temp.add(root.val);
map.put(curVal,temp);
}
else {
List<Integer> tempList = new ArrayList<>();
tempList.add(root.val);
map.put(curVal,tempList);
}
dfs(root.left,curVal - 1);
dfs(root.right,curVal + 1);
}
}
解决方法
否定键没有问题,但是缺少一些东西。
-
HashMap不会保持任何顺序,即使这样做,您也将按深度优先顺序将元素添加到HashMap中,这意味着首先要添加根。因此,在输出中没有得到预期的订单是很正常的。
您应该改为按键值的顺序对地图进行迭代。
-
将多个值添加到同一地图条目(因为键相同)时,您还将按照访问它们的顺序添加它们。但是,不能保证此顺序是所需的顺序:深度优先迭代将在树中上下移动,因此不能保证您先处理较高的节点,然后再处理较低的节点。如果需要,则需要广度优先。另外,您可以将y坐标(水平)作为参数传递。
-
当两个节点的x和y坐标相同时,应按升序报告其值。您对此没有任何准备。如果您坚持深度优先,那么在地图条目中,您将需要另一张以y坐标为键的地图。然后,当您发现当前y坐标已经有一个条目时,应确保以正确的顺序添加当前节点的值。请注意,这会为您的数据结构添加一个嵌套级别。最好先添加值而不进行排序,然后在算法的最后阶段遍历完整的数据结构并对所有这些小子数组进行排序。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。