두 갈래 트리와 k의 모든 경로 찾기 Path Sum II

1418 단어
제목: Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.이 문제는 이전의 경로와 문제에 비해 한 단계 향상되었다.이전의 문제는 이런 경로가 있는지 판단하고 볼 값을 되돌려주면 된다는 것이다.이 문제는 모든 경로를 찾아내고 되돌려 주는 것이다.
주의: 두 갈래 나무의 결점 원소는 음수일 수 있으며, 잎 노드에 가지 않으면 경로와 k인지 영원히 알 수 없습니다.
사고방식: 차례차례 사고방식.모든 경로를 깊이 파고들어 잎사귀 노드로 가서 판단한다.
코드:
/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int> > pathSum(TreeNode *root, int sum) {
        vector<vector<int> > v;
        vector<int> tmp;
        if(root == NULL)
            return v;
        findpath(root, sum, v, tmp);
        return v;
    }
    
    void findpath(TreeNode * root, int sum, vector<vector<int> > &v, vector<int> tmp)
    {
        if(root->left == NULL && root->right == NULL && sum == root->val)
        {
            tmp.push_back(sum);
            v.push_back(tmp);
            return;
        }
        
        tmp.push_back(root->val);
        if(root->left != NULL)
            findpath(root->left, sum-root->val, v, tmp);
        if(root->right != NULL)
            findpath(root->right, sum-root->val, v, tmp);
    }
};

좋은 웹페이지 즐겨찾기