【力扣】112. 路径总和

递归、回溯法:

class Solution {
    private:
    bool traversal(TreeNode* cur,int targetSum){
        //终止条件:
        if(!cur->left&&!cur->right&&targetSum==0) return true;
        if(!cur->left&&!cur->right) return false;
        //递归:
        if(cur->left) {
            targetSum=targetSum-cur->left->val;//递归
            if(traversal(cur->left,targetSum))
            return true;
            targetSum=targetSum+cur->left->val;//回溯
        }//左
        if(cur->right) {
            targetSum=targetSum-cur->right->val;//递归
            if(traversal(cur->right,targetSum))
            return true;
            targetSum=targetSum+cur->right->val;//回溯
        }//右
        return false;
    }
public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root==nullptr) return false;
        return traversal(root,targetSum-root->val);//根    
    }
};

递归【精简版】

class Solution {
public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root==nullptr) return false;
         //终止条件:
        if(!root->left&&!root->right&&targetSum==root->val) return true;
        return hasPathSum(root->left,targetSum-root->val)||hasPathSum(root->right,targetSum-root->val);    
    }
};

迭代法:

  • 使用栈实现迭代法
  • 前序遍历,入栈的时候先右后左
  • 定义pair数据类型,既记录该节点又记录路径和
  • pair:first、second
class Solution {
public:
    bool hasPathSum(TreeNode* root, int targetSum) {
        if(root==nullptr) return false;
        stack<pair<TreeNode*,int>> st;//使用栈实现迭代法,栈中数据需要保留该节点以及路径和
        st.push(pair<TreeNode*,int> (root,root->val));//根
        while(!st.empty()){
            pair<TreeNode*,int> node=st.top();
            st.pop();
            if(node.first->left==nullptr&&node.first->right==nullptr&&node.second==targetSum) return true;
            //进栈:根->右->左;出栈:根->左->右
            if(node.first->right){
                st.push(pair<TreeNode*,int>(node.first->right,node.second+node.first->right->val));//右
            }
            if(node.first->left){
                st.push(pair<TreeNode*,int>(node.first->left,node.second+node.first->left->val));//左
            }
        }
        return false;
    }
};

你可能感兴趣的:(力扣,leetcode,算法,c++)