按之字形程序打印二叉树
题目形容
请实现一个函数依照之字形打印二叉树,即第一行依照从左到右的程序打印,第二层依照从右至左的程序打印,第三行依照从左到右的程序打印,其余行以此类推。
题目链接 : 按之字形程序打印二叉树
代码
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Stack;
/**
* 题目:按之字形程序打印二叉树
* 题目形容
* 请实现一个函数依照之字形打印二叉树,即第一行依照从左到右的程序打印,第二层依照从右至左的程序打印,第三行依照从左到右的程序打印,其余行以此类推。* 题目链接:* https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0?tpId=13&&tqId=11212&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
*/
public class Jz59 {public ArrayList<ArrayList<Integer>> print(TreeNode pRoot) {Queue<TreeNode> treeNodes = new LinkedList<>();
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
treeNodes.add(pRoot);
boolean flag = true; // 当 flag 为 true 时,栈从左向右放;当 flag 为 false 时,栈从右向左放
while (!treeNodes.isEmpty()) {int cnt = treeNodes.size();
ArrayList<Integer> curList = new ArrayList<>();
Stack<TreeNode> curNodes = new Stack<>();
while (cnt-- > 0) {TreeNode t = treeNodes.poll();
if (t == null) {continue;}
curList.add(t.val);
if (flag) {curNodes.add(t.left);
curNodes.add(t.right);
} else {curNodes.add(t.right);
curNodes.add(t.left);
}
}
flag = !flag;
while (!curNodes.isEmpty()) {treeNodes.add(curNodes.pop());
}
if (!curList.isEmpty()) {result.add(curList);
}
}
return result;
}
public static void main(String[] args) {}}
【每日寄语】只有你明天再多致力一下,那个将来能够像星星一样闪闪发光的人就是你呀!