【Lintcode】1577. Sum of Leaf Nodes

题目地址：

https://www.lintcode.com/problem/1577/description

求一棵二叉树中所有叶子节点的和。要求空间$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(1)$。