题目地址:
https://www.lintcode.com/problem/1577/description
求一棵二叉树中所有叶子节点的和。要求空间 O ( 1 ) O(1) O(1)。
空间要求 O ( 1 ) O(1) O(1)的话只能Morris遍历了。由于Morris遍历会将二叉树线索化,所以这里的关键问题是怎么找到所有的叶子。把所有的叶子分为两类,一类在左链里,另一类在右链里(这里”在左链“里意思是它有父亲,且是其父亲的左孩子,不”在左链“就定义其在右链里)。由于在左链里的叶子会线索化(即其右指针会指向其后继),所以要在线索化的时候将其加入答案,这样下次走到它的时候,由于它被线索化了,右指针非空,从而不会被重复计算;对于在右链里的叶子,它不会线索化,可以在第一次遍历到它的时候加入答案。代码如下:
public class Solution {
/**
* @param root:
* @return: the sum of leafnode
*/
public int sumLeafNode(TreeNode root) {
// Write your code here
int res = 0;
TreeNode cur = root;
while (cur != null) {
// 将右链的叶子加入答案
if (cur.left == null && cur.right == null) {
res += cur.val;
}
TreeNode left = cur.left;
if (left == null) {
cur = cur.right;
} else {
while (left.right != null && left.right != cur) {
left = left.right;
}
// 将左链的叶子加入答案
if (left.left == null && left.right == null) {
res += left.val;
}
if (left.right == null) {
left.right = cur;
cur = cur.left;
} else {
left.right = null;
cur = cur.right;
}
}
}
return res;
}
}
class TreeNode {
int val;
TreeNode left, right;
public TreeNode(int val) {
this.val = val;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)。