二叉樹的路徑和

問題描述:

給定一個二叉樹,找出所有路徑中各節點相加總和等於給定目標值 的路徑。

一個有效的路徑,指的是從根節點到葉節點的路徑。

樣例

給定一個二叉樹,和 目標值 = 5:

     1
    / \
   2   4
  / \
 2   3

返回:

[
  [1, 2, 2],
  [1, 4]
]

解題思路:

建立一個二維向量,用以儲存符合要求的路徑,路徑表示爲一維向量的形式。首先判斷根節點的左右子樹是否爲空,將根節點的數值放入一維向量中,如果左右子樹都爲空判斷根節點的數值是否等於給定的數值,若等於給定的數值,則將其放入二維向量中。定義一個整數型t,用以計算二叉樹路徑的長度。如果左子樹不爲空,則將左子樹節點的數值加在t上,同時將左子樹節點放入一維向量中,然後繼續判斷,繼續調用函數,直到葉子節點,即左右子樹都爲空,然後判斷路徑長度是否等於給定的數值,若滿足條件,將這條路徑對應的一維向量放入二維向量中。對於右子樹的情況做相同操作。

代碼實現:

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root the root of binary tree
     * @param target an integer
     * @return all valid paths
     */
    vector<vector<int>>lz;
    vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
        // Write your code here
        if(root==NULL)  
            return lz;  
        vector<int>l;  
        func(root,root->val,target,l);  
        return lz;

    }
    void func(TreeNode *root,int t,int target,vector<int> l){  
        l.push_back(root->val);
        if(root->left==NULL&&root->right==NULL){  
            if(t==target)  
            lz.push_back(l);  
            return;  
        }  
        int s;  
         if(root->left!=NULL){  
            s=t+root->left->val;  
            func(root->left,s,target,l);  
        }  
        if(root->right!=NULL){  
             s=t+root->right->val;  
             func(root->right,s,target,l);  
        }  
    }  
};   

解題感悟:

需要在給定的函數中調用一個新的函數,新函數寫在給定函數體外,若直接在給定的函數體內寫操作代碼,會出現restricted call錯誤。錯誤情況如下:

      if(root==NULL) return lz;
        else{
            vector<int>l;int t=root->val;
            l.push_back(root->val);
            if(root->left==NULL&&root->right==NULL&&t==target){
                lz.push_back(l);
            return lz;
            }
            if(root->left!=NULL){
                t+=root->left->val;
                binaryTreePathSum(root->left,target);
            }
            if(root->right!=NULL){
                t+=root->right->val;
                binaryTreePathSum(root->right,target);
            }
            
        }
        
    }
};

輸入
查看如何表示一棵 二叉樹
{1,2,4,2,3}, 5
期望答案
[[1,2,2],[1,4]]
錯誤信息

TERMSIG= 31

提示
你的代碼造成了一些危險行爲,例如非法系統調用。Restricted call通常是由引用了不存在的數組下標造成的


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章