zoukankan      html  css  js  c++  java
  • LeetCode113 Path Sum II

    Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.(Medium)

    For example:
    Given the below binary tree and sum = 22,

                  5
                 / 
                4   8
               /   / 
              11  13  4
             /      / 
            7    2  5   1
    

    return

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

    分析:

    回溯法处理即可,利用helper函数遍历所有路径,达到sum时添加到result里。

    代码:

     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     vector<vector<int>> result;
    12     void helper(TreeNode* root, int sum, vector<int>& vec) {
    13         if (root == nullptr) {
    14             return;
    15         }
    16         vec.push_back(root -> val);
    17         sum -= root -> val;
    18         if (root -> left == nullptr && root -> right == nullptr) {
    19             if (sum == 0) {
    20                 result.push_back(vec);
    21             }
    22             else {
    23                 vec.pop_back();
    24                 sum += root -> val;
    25                 return;
    26             }
    27         }
    28         if (root -> left != nullptr) {
    29             helper(root -> left, sum, vec);
    30         }
    31         if (root -> right != nullptr) {
    32             helper(root -> right, sum, vec);
    33         }
    34         vec.pop_back();
    35         sum -= root -> val;
    36     }
    37 public:
    38     vector<vector<int>> pathSum(TreeNode* root, int sum) {
    39         vector<int> vec;
    40         helper(root, sum, vec);
    41         return result;
    42     }
    43 };
  • 相关阅读:
    三维几何模板
    凸包
    计算几何 部分模板
    几何模板
    高斯消元模板
    tarjin求割点
    在无向图中找最短桥(tarjan)
    线段树
    错排问题
    NABCD分析
  • 原文地址:https://www.cnblogs.com/wangxiaobao/p/6071803.html
Copyright © 2011-2022 走看看