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

最新推荐文章于 2024-03-23 22:42:54 发布

春树暮云_

最新推荐文章于 2024-03-23 22:42:54 发布

阅读量542

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/Capricorn_devil/article/details/89027920

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;
    }