与第二章类似,LeetCode指定了TreeNode实现。为了方便后续习题的编写和测试,创建一个基础父类,包含TreeNode实现,以及create()和print()创建和打印树的方法。其中create()采用类似“堆”的方式,用数组表示树形结构,复习一下,左右子结点就是2*i和2*i+1。而print()方法采用前序遍历的方式,通过额外一个参数level确定当前结点的深度,从而打印一些制表符或其他符号来表示出结点的父子关系。两个核心方法都使用递归的方式,实现起来非常简洁!
注:实现时一切都感觉很良好,结果又犯了一个低级错误。在print()中最初写成了
while(level--) cout << "\t";
结果忘记level值被修改了(变成了-1),导致后面递归调用时level+1就始终是0了。
上面的工具函数使用起来很方便,创建一个Solution子类。唯一缺点就是打印时没有指出子结点到底是左孩子还是右孩子,在部分结点缺失的情况下看起来稍稍有点乱。
时间: 2024-10-03 01:55:12