leetcode--113.총 경로 II

3459 단어 leetCode
두 갈래 나무와 목표와 뿌리 노드에서 잎 노드까지의 모든 경로를 찾는 것은 목표와 같은 경로입니다.
설명: 잎 노드는 하위 노드가 없는 노드를 가리킨다.
예: 다음과 같은 두 갈래 트리와 목표 및sum = 22 ,
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

반환:
[
   [5,4,11,2],
   [5,8,4,5]
]
// :( )
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    vector> ans;
    vector temp;
public:
    vector> pathSum(TreeNode* root, int sum) {
        if(root==NULL) return ans;
        helper(root,sum, root->val);
        return ans;
                
    }
    
    void helper(TreeNode* root, int sum, int s)
    {
        if(root==NULL) return;
        sum -= root->val;
        temp.push_back(root->val);
        if(root->left == NULL && root->right == NULL)
        {
            if(sum == 0)
            {
                ans.push_back(temp);
                temp.clear();
                temp.push_back(s);                
            }else if(sum < 0)
            {
                temp.pop_back();
                return;
            }
        }            
        if(root->left) helper(root->left,sum,s);
        if(root->right) helper(root->right,sum,s);
        return;
    }
};

 
대장부 정확한 코드:
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector> pathSum(TreeNode* root, int sum) {
        vector v;
        vector> vv;
        DFS(vv,v,root,sum);
        return vv;
    }
    void DFS(vector> &vv,vector v,TreeNode* root,int sum){
        if(!root)
            return;
        sum = sum-root->val;
        v.push_back(root->val);
        if(!sum && !root->left && !root->right)
            vv.push_back(v);
        DFS(vv,v,root->left,sum);
        DFS(vv,v,root->right,sum);
    }
};

수정된 올바른 코드:
 

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    vector> ans;
    vector temp;
public:
    vector> pathSum(TreeNode* root, int sum) {
        if(root==NULL) return ans;
        helper(root,sum);
        return ans;
                
    }
    
    void helper(TreeNode* root, int sum)
    {
        if(root==NULL) return;
        sum -= root->val;
        temp.push_back(root->val);
        if(root->left == NULL && root->right == NULL)
        {
            if(sum == 0)
            {
                ans.push_back(temp);              
            }
        }            
        if(root->left) helper(root->left,sum);
        if(root->right) helper(root->right,sum);
        temp.pop_back();  // 
        return;
    }
};

요약:
이 문제는 거슬러 올라가야 한다. 거슬러 올라가는 것은 귀속 뒤에 두어야 한다.

좋은 웹페이지 즐겨찾기