二叉树的垂序遍历(基于前序遍历的升华)

75 篇文章 0 订阅
15 篇文章 0 订阅

在这里插入图片描述
核心思路:
自定义排序
每个节点有三个属性:值val,行坐标row,列坐标col
1、先按列排序,从小到大;

2、列相同,再按行排序,从小到大;

3、行相同,按值的大小排序,从小到大。
所以我们有两种实现思路:用双哈希表或者结构体自定义排序
双哈希表:
一个表存储列,一个表存储行,用数组每次记录数据,然后对数组进行自定义排序

/**
 * 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 {
      Map<Integer,Integer> colmap=new HashMap<>();
      Map<Integer,Integer> rowmap=new HashMap<>();
       List<Integer> path=new ArrayList<>();
    public List<List<Integer>> verticalTraversal(TreeNode root) {
      dfs(root,0,0);
      List<List<Integer>> ans=new ArrayList<>();
       Collections.sort(path,new Comparator<Integer>(){
          @Override
          public int compare(Integer a,Integer b){
              if(colmap.get(a)==colmap.get(b)){
                  if(rowmap.get(a)==rowmap.get(b))return a-b;
                  return rowmap.get(a)-rowmap.get(b);
              }
              return colmap.get(a)-colmap.get(b);
          }
      });
      List<Integer> res=new ArrayList<>();
      res.add(path.get(0));
      for(int i=1;i<path.size();i++){
          int ch=path.get(i);
          int chl=path.get(i-1);
          if(colmap.get(ch)!=colmap.get(chl)){
              ans.add(res);
              res=new ArrayList<>();
          }
           res.add(ch);
      }
      ans.add(res);
      return ans;
    }
    public void dfs(TreeNode root,int row,int col){
        if(root==null)return;
         path.add(root.val);
         rowmap.put(root.val,row);
         colmap.put(root.val,col);
         dfs(root.left,row+1,col-1);
         dfs(root.right,row+1,col+1);
    }
}

或者用treemap

/**
 * 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 {
      Map<Integer,List<Integer>> map=new TreeMap<>();
      Map<Integer,Integer> ceq=new HashMap<>();
    public List<List<Integer>> verticalTraversal(TreeNode root) {
      dfs(root,0,0);
      List<List<Integer>> ans=new ArrayList<>();
      for(int ch:map.keySet()){
           List<Integer> path=map.get(ch);
           Collections.sort(path,new Comparator<Integer>(){
              @Override
              public int compare(Integer a,Integer b){
                  if(ceq.get(a)==ceq.get(b))return a-b;
                  return ceq.get(a)-ceq.get(b);
              }
           });        
           ans.add(path);
      }
      return ans;
    }
    public void dfs(TreeNode root,int row,int col){
        if(root==null)return;
         List<Integer> path=map.getOrDefault(col,new ArrayList<>());
         path.add(root.val);
         ceq.put(root.val,row);
         map.put(col,path);
         dfs(root.left,row+1,col-1);
         dfs(root.right,row+1,col+1);
    }
}

结构体:
思想跟上述一样:

/**
 * 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 {
    class Node{
     int val;
     int col;
     int row;
     Node(int x){
         this.val=x;
     }
    }
    List<Node> path=new ArrayList<>();
    public List<List<Integer>> verticalTraversal(TreeNode root) {
      dfs(root,0,0);
      List<List<Integer>> ans=new ArrayList<>();
      Collections.sort(path,new Comparator<Node>(){
          @Override
          public int compare(Node a,Node b){
              if(a.col==b.col){
                  if(a.row==b.row)return a.val-b.val;
                  return a.row-b.row;
              }
              return a.col-b.col;
          }
      });
      List<Integer> res=new ArrayList<>();
      res.add(path.get(0).val);
      for(int i=1;i<path.size();i++){
          Node ch=path.get(i);
          Node chl=path.get(i-1);
          if(ch.col!=chl.col){
              ans.add(res);
              res=new ArrayList<>();
          }
          res.add(ch.val);
      }
      ans.add(res);
      return ans;
    }
    public void dfs(TreeNode root,int row,int col){
        if(root==null)return;
         Node ch=new Node(root.val);
        ch.row=row;
        ch.col=col;
          path.add(ch);
         dfs(root.left,row+1,col-1);
         dfs(root.right,row+1,col+1);
    }
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
根据二叉树的中序遍历和后序遍历输出前序遍历的方法如下: 1. 首先,我们知道前序遍历的顺序是:根节点 -> 左子树 -> 右子树。 2. 后序遍历的顺序是:左子树 -> 右子树 -> 根节点。 3. 中序遍历的顺序是:左子树 -> 根节点 -> 右子树。 根据以上信息,我们可以得出以下步骤来输出前序遍历: 1. 后序遍历的最后一个节点一定是根节点。 2. 在中序遍历中找到根节点的位置,将中序遍历分为左子树和右子树。 3. 根据中序遍历中左子树和右子树的节点数量,将后序遍历分为左子树和右子树。 4. 递归地对左子树和右子树进行相同的操作,直到只剩下一个节点。 5. 输出当前节点的值,然后递归地输出左子树和右子树的节点值。 下面是一个示例代码来实现这个过程: ```python def build_preorder(inorder, inorder_start, inorder_end, postorder, postorder_start, postorder_end): if inorder_start > inorder_end: return [] root_val = postorder[postorder_end] root_index = inorder.index(root_val) left_size = root_index - inorder_start right_size = inorder_end - root_index left_preorder = build_preorder(inorder, inorder_start, root_index - 1, postorder, postorder_start, postorder_start + left_size - 1) right_preorder = build_preorder(inorder, root_index + 1, inorder_end, postorder, postorder_end - right_size, postorder_end - 1) return [root_val] + left_preorder + right_preorder def get_preorder(inorder, postorder): return build_preorder(inorder, 0, len(inorder) - 1, postorder, 0, len(postorder) - 1) # 示例输入 inorder = [4, 2, 5, 1, 6, 3, 7] postorder = [4, 5, 2, 6, 7, 3, 1] preorder = get_preorder(inorder, postorder) print(preorder) ``` 输出结果为:[1, 2, 4, 5, 3, 6, 7]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值