• 剑指34 二叉树中和为某一值的路径


    输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

    示例:
    给定如下二叉树,以及目标和 sum = 22,

    5
    /
    4 8
    / /
    11 13 4
    / /
    7 2 5 1
    返回:

    [
    [5,4,11,2],
    [5,8,4,5]
    ]
     

    提示:

    节点总数 <= 10000

    这题采用递归,如果到了叶节点同时路径和与目标相等就可以输出。但是需要注意的是,题目没有限制目标值和节点值的正负,所以节点可以有正有负,在没有到达叶子结点之前不能提前判断该路径合不合法。此外注意处理根节点为空指针的情况。

     1 /**
     2  * Definition for a binary tree node.
     3  * struct TreeNode {
     4  *     int val;
     5  *     TreeNode *left;
     6  *     TreeNode *right;
     7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     8  * };
     9  */
    10 class Solution {
    11 public:
    12     vector<vector<int>> ret;
    13     vector<vector<int>> pathSum(TreeNode* root, int sum) {
    14         if(root==nullptr)
    15             return ret;
    16         recursive_find({},root,sum,0);
    17         return ret;
    18     }
    19 
    20     void recursive_find(vector<int> partial_ret, TreeNode* curnode,int target, int cursum){
    21         if(curnode->left==nullptr && curnode->right==nullptr){
    22             if(curnode->val==target-cursum){
    23                 partial_ret.push_back(curnode->val);
    24                 ret.push_back(partial_ret);
    25                 return ;
    26             }
    27             else
    28                 return;
    29         }
    30         cursum+=curnode->val;
    31         partial_ret.push_back(curnode->val);
    32         if(curnode->left!=nullptr)
    33             recursive_find(partial_ret,curnode->left,target,cursum);
    34         if(curnode->right!=nullptr)
    35             recursive_find(partial_ret,curnode->right,target,cursum);
    36         return ;
    37     }
    38 
    39 };

    这里传入的partial_ret是传值,这样好处是返回前不用做pop_back。书上的做法是传引用,这样的好处是传的只是一个指针,而且不用多次复制,所以递归时节省了栈的空间和复制的时间,这么看来传引用更加高效。

  • 相关阅读:
    vs2005&sql2005
    根据老外的方法,GridVew实现多列排序.
    @好处多多
    系统启动后,自动启动软件方法.
    A4和B5纸张大小是按什么标准划分的?
    easyui控件的加载顺序
    jQuery模拟点击A标记
    求阶乘
    文件上传
    Andriod:环境搭建与配置
  • 原文地址:https://www.cnblogs.com/rookiez/p/13234938.html
Copyright © 2020-2023  润新知