112. 路径总和（非递归方法）

原创于 2019-04-04 16:44:44 发布 · 542 阅读

1 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

11 篇文章

订阅专栏

这篇博客探讨了如何使用非递归方法解决路径总和的问题，关键点在于在回退时判断并减去节点值，特别是当右子树遍历完毕时。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

 bool hasPathSum(TreeNode* root, int sum) {
         if (root == NULL) 
            return false;
         if (root->val == sum && root->left ==  NULL && root->right == NULL) 
            return true;
         return hasPathSum(root->left, sum-root->val) || hasPathSum(root->right, sum-root->val);
    }

非递归的关键在于回退时在什么情况下减去节点的值，即该节点右子树已经遍历，要减去该节点值，回退到上一层

bool hasPathSum(TreeNode *root, int sum) {
        stack<TreeNode *> s;
        TreeNode *pre = NULL, *cur = root;
        int SUM = 0;
        while (cur || !s.empty()) {
            while (cur) {
                s.push(cur);
                SUM += cur->val;
                cur = cur->left;
            }
            cur = s.top();
            if (cur->left == NULL && cur->right == NULL && SUM == sum) {
                return true;
            }
            if (cur->right && pre != cur->right) {
                cur = cur->right;
            } else {
                pre = cur;
                s.pop();
                SUM -= cur->val;
                cur = NULL;
            }
        }
        return false;
    }