JAVA程序设计：从先序遍历还原二叉树（LeetCode：1028）

我们从二叉树的根节点 root 开始进行深度优先搜索。

在遍历中的每个节点处，我们输出 D 条短划线（其中 D 是该节点的深度），然后输出该节点的值。（如果节点的深度为 D，则其直接子节点的深度为 D + 1。根节点的深度为 0）。

如果节点只有一个子节点，那么保证该子节点为左子节点。

给出遍历输出 S，还原树并返回其根节点 root。

 

示例 1：

输入："1-2--3--4-5--6--7"
输出：[1,2,5,3,4,6,7]
示例 2：

输入："1-2--3---4-5--6---7"
输出：[1,2,5,3,null,6,null,4,null,7]
示例 3：

输入："1-401--349---90--88"
输出：[1,401,null,349,88,90]
 

提示：

原始树中的节点数介于 1 和 1000 之间。
每个节点的值介于 1 和 10 ^ 9 之间。

思路：我们定义两个队列分别用来存储这个节点属于那一层以及这个节点的值，而队列里的顺序就是先序遍历的顺序，后面就是模拟先序遍历啦。

class Solution {
	
	TreeNode root;
	private Queue<Integer> q1,q2;
	
    public TreeNode recoverFromPreorder(String S) {
    	
    	int num=0,sum=0;
    	int n=S.length();
    	q1=new LinkedList<>();
    	q2=new LinkedList<>();
    	
    	for(int i=0;i<n;i++) {
    		if(S.charAt(i)>='0' && S.charAt(i)<='9') {
    			sum=sum*10+(S.charAt(i)-'0');
    			if(i+1==n || S.charAt(i+1)=='-') {
        			q1.add(num);
        			q2.add(sum);
        			sum=num=0;
        		}
    		}
    		else
    			num++;
    	}
    	
    	q1.poll();
    	root=new TreeNode(q2.poll());
    	
    	TreeNode ans=root;
    	
    	dfs(root,0);
    	
    	return ans;
    }
    
    private void dfs(TreeNode root,int depth) {
    	
    	if(root==null)
    		return;
    	
    	if(!q1.isEmpty() && q1.peek()==depth+1) {
    		root.left=new TreeNode(q2.poll());
    		q1.poll();
    	}
    	
    	dfs(root.left,depth+1);
    	
    	if(!q1.isEmpty() && q1.peek()==depth+1) {
    		root.right=new TreeNode(q2.poll());
    		q1.poll();
    	}

    	dfs(root.right,depth+1);
    	
    }
}