Day 44
题目描述
思路
这题类似与昨天那题,首先来复习一下,后序遍历,对于后序遍历每一个元素都满足以下规律:
(左子树)(右子树)(根),那么我们直接修改昨天的代码即可。前序是从前向后找根,后序我们就从后向前找根。
代码如下:
/*** 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 Map<Integer,Integer>indexmap;public TreeNode findroot(int[]inorder,int[]postorder,int inleft,int inright,int postleft,int postright){if(postleft>postright){return null;}int root_num=postorder[postright];//从后向前找根int in=indexmap.get(postorder[postright]);//获取根在中序遍历中的序号TreeNode root=new TreeNode(root_num);root.left=findroot(inorder,postorder,inleft,in-1,postleft,postright-inright+in-1);root.right=findroot(inorder,postorder,in+1,inright,postright-inright+in,postright-1);return root;}public TreeNode buildTree(int[] inorder, int[] postorder) {indexmap=new HashMap<Integer,Integer>();int n=inorder.length;for(int i=0;i<n;i++){//存放每个元素在中序遍历中的序号indexmap.put(inorder[i],i);}TreeNode root=findroot(inorder,postorder,0,n-1,0,n-1);return root;}
}