Java教程

112.path-sum 路径总和

本文主要是介绍112.path-sum 路径总和,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

带明显的回溯的版本

#include <vector>
using std::vector;
class Solution {
  private:
    vector<int> res;
    int sum = 0;

  public:
    void cnt_sum(TreeNode *root) {
        if (root->left == nullptr && root->right == nullptr) {
            sum += root->val;
            res.push_back(sum);
            return;
        }
        sum += root->val;
        if (root->left != nullptr) {
            cnt_sum(root->left);
            sum -= root->left->val;//回溯
        }
        if (root->right != nullptr) {
            cnt_sum(root->right);
            sum -= root->right->val;//回溯
        }
        return;
    }
    bool hasPathSum(TreeNode *root, int targetSum) {
        if (root == nullptr)
            return false;
        cnt_sum(root);
        for (int i : res) {
            if (i == targetSum)
                return true;
        }
        return false;
    }
};

回溯被隐藏起来的版本

#include <vector>
using std::vector;
class Solution {
  private:
    vector<int> res;
    //int sum = 0;

  public:
    void cnt_sum(TreeNode *root, int sum) {
        if (root->left == nullptr && root->right == nullptr) {
            sum += root->val;
            res.push_back(sum);
            return;
        }
        sum += root->val;
        if (root->left != nullptr) {
            cnt_sum(root->left, sum); //注意这里的sum作为实参,其值被改变并不会改变实际上sum的值
            //sum -= root->left->val;
        }
        if (root->right != nullptr) {
            cnt_sum(root->right, sum);
            //sum -= root->right->val;
        }
        return;
    }
    bool hasPathSum(TreeNode *root, int targetSum) {
        if (root == nullptr)
            return false;
        cnt_sum(root, 0);
        for (int i = 0; i < res.size(); i++)
            if (res[i] == targetSum)
                return true;
        return false;
    }
};
这篇关于112.path-sum 路径总和的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!