树形结构是计算机科学中一种经典的数据结构。从前端编程当中UI界面层级结构,到java中的HashMap,到MySQL中的数据存储,到操作系统底层文件系统等等,这些都涉及到树形结构。
树形结构的基本操作就是遍历,初学者可能会问:遍历就是遍历为什么还要分前中后三种方法呢?然而,要回答这个问题就要涉及到树形结构的更多知识了,比如:如果给定的树是一棵‘二叉搜索树’,那么如果中序遍历则会按从小大的顺序遍历这个树。再比如:用程序实现递归删除一个文件夹包括里面的文件,也就是说要删除一个文件夹必须要保证文件夹下的所有文件和文件夹全部被删除,这种是典型的后序遍历方式。
二叉树的结构是一个节点包含一个左子节点和一个右子节点。也就是:父亲节点、左孩子节点、右孩子节点,为了表述方便我们把它们称为:父节点、左节点、右节点。
为了方便描述我们先把节点结构TreeNode类给出:
public class TreeNode
{
public int val;
public TreeNode left;
public TreeNode right;
public TreeNode(int x) { val = x; }
}
二叉树-前序遍历
前序遍历的顺序为:父节点、左节点、右节点。也就是我们再访问一个节点的时候,首先要访问父节点,然后依次访问左节点和右节点,三个节点都访问完成为一次遍历。话不多说直接上代码:
public IList PreorderTreewalk(TreeNode root){
var result = new List();
var stack = new Stack();
stack.Push(root);
while (stack.Count > 0){
var n = stack.Pop();
result.Add(n.val);
if (n.right != null)
stack.Push(n.right);
if (n.left != null)
stack.Push(n.left);
}
return result;
}
核心代码就是while循环中的几行,主要分两步:
1.每次从栈中弹出一个元素,然后add进result数组。2.把当前元素的孩子入栈。
二叉树-中序遍历
中序遍历的顺序为:左节点、父节点、右节点。中序遍历比前序遍历稍微复杂点,因为它不像前序遍历每次循环开始栈顶元素就是要访问的元素,而是先访问其左孩子。
public IList InorderTreewalk(TreeNode root){
var node = root;
var result = new List();
var stack = new Stack();
while(node != null || stack.Count > 0){
while(node != null){
stack.Push(node);
node = node.left;
}
node = stack.Pop();
result.Add(node.val);
node = node.right;
}
}
二叉树-后序遍历
中序遍历的顺序为:左节点、右节点、父节点。
public IList PostorderTreewalk(TreeNode root){
var node = root;
var result = new List();
var stack = new Stack();
TreeNode last = null;
while (node != null || stack.Count > 0){
while (node != null){
stack.Push(node);
node = node.left;
}
node = stack.Peek();
if (node.right == null || node.right == last){
result.Add(node.val);
last = node;
node = null;
stack.Pop();
}else{
node = node.right;
}
}
return result;
}后序遍历 - 前序遍历改进版
public IList PostorderTreewalk(TreeNode root){
var result = new List();
var stack = new Stack();
stack.Push(root);
while (stack.Count > 0){
var n = stack.Pop();
result.Insert(0, n.val);
if (n.left != null)
stack.Push(n.left);
if (n.right != null)
stack.Push(n.right);
}
return result;
}