algorithm-class/算法周更班/class_2023_02_3_week/Code02_RecoverTreeFromPreorderTraversal.java

package class_2023_02_3_week;

// 我们从二叉树的根节点 root 开始进行深度优先搜索。
// 在遍历中的每个节点处，我们输出 D 条短划线（其中 D 是该节点的深度）
// 然后输出该节点的值。（如果节点的深度为 D，则其直接子节点的深度为 D + 1
// 根节点的深度为 0
// 如果节点只有一个子节点，那么保证该子节点为左子节点
// 给出遍历输出 S，还原树并返回其根节点 root。
// 测试链接 : https://leetcode.cn/problems/recover-a-tree-from-preorder-traversal/
public class Code02_RecoverTreeFromPreorderTraversal {

	// 不提交这个类
	public static class TreeNode {
		public int val;
		public TreeNode left;
		public TreeNode right;

		public TreeNode(int v) {
			val = v;
		}
	}

	// 提交如下的代码
	public static int MAXN = 2001;

	public static int[] queue = new int[MAXN];

	public static int l, r;

	public static TreeNode recoverFromPreorder(String traversal) {
		l = 0;
		r = 0;
		int number = 0;
		int level = 0;
		boolean pickLevel = true;
		for (int i = 0; i < traversal.length(); i++) {
			if (traversal.charAt(i) != '-') {
				if (pickLevel) {
					queue[r++] = level;
					level = 0;
					pickLevel = false;
				}
				number = number * 10 + traversal.charAt(i) - '0';
			} else {
				if (!pickLevel) {
					queue[r++] = number;
					number = 0;
					pickLevel = true;
				}
				level++;
			}
		}
		queue[r++] = number;
		return f();
	}

	// 当前，消费的时间是 :
	// 层 : queue[l]
	// 节点值 : queue[l+1]
	public static TreeNode f() {
		int level = queue[l++];
		TreeNode head = new TreeNode(queue[l++]);
		if (l < r && queue[l] > level) {
			head.left = f();
		}
		if (l < r && queue[l] > level) {
			head.right = f();
		}
		return head;
	}

}