在看《小灰算法》时,练习代码,其中一小段代码,了解了while和if的区别,先说结论:
while如果判断为true,执行代码块,然后会继续判断条件,如果条件还是true,还是会执行代码块,直到判断条件为false。
if如果判断为true,执行代码块,执行完后不会在进行条件判断,去执行其他代码。
举例说明:
二叉树如下:
利用栈前序遍历二叉树,应该输出:3-2-9-10-8-4
我写的代码如下:
/**
* @Description 利用栈前序遍历二叉树
* @Param [args]
*/
public static void preOrderTraveralWithStack(TreeNode root){
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode treeNode = root;
while (treeNode!=null || !stack.isEmpty()){
//迭代访问节点的左孩子,并入栈
while (treeNode != null){
System.out.println(treeNode.data);
stack.push(treeNode);
treeNode = treeNode.leftChild;
}
//如果节点没有左孩子,则弹出栈顶节点,访问节点右孩子
while (!stack.isEmpty()){
treeNode = stack.pop();
treeNode = treeNode.rightChild;
}
}
}
输出结果3-2-9-8-4
原因是第二个while出了问题,在节点9取出左节点为空时,进入第二个while判断,然后while(!stack.isEmpty())一直为true,直到stack为空,这时候栈被吐完,3的右节点为8,10已经过去了,所以永远不会再输出10了
解决,将while(!stack.isEmpty())改为if(!stack.isEmpty())
以上