LeetCode 437. Path Sum III 문제 및 고정된 트리 경로 수

3360 단어 LeetCode
You are given a binary tree in which each node contains an integer value.
Find the number of paths that sum to a given value.
The path does not need to start or end at the root or a leaf, but it must go downwards (traveling only from parent nodes to child nodes).
The tree has no more than 1,000 nodes and the values are in the range -1,000,000 to 1,000,000. root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8
  10
 /  \
5   -3
/ \    \
3   2   11
/ \   \
3  -2   1

Return 3. The paths that sum to 8 are:
  • 5 -> 3
  • 5 -> 2 -> 1
  • -3 -> 11

  • 분석: 1.전체 트리를 훑어보고 노드마다 고정된 경로 수량을 찾습니다.2. 귀속 연산을 사용하여 두 갈래 트리 경로와 계산하기;2. 아이 값 사이를 계산하고 임시 경로와temp를 설정하여 다른 아이를 찾을 때 이전 경로의 것과 변하지 않도록 확보한다(현재 경로와sum, 경로 총수를 만족시키는 cnt를 모두 참고하여 반환할 때 수정된sum와 cnt를 동시에 전달하는 방법을 몰라서 임시 값으로 대체했다).3. 경로와 목표가 일치하는지 판단할 때 일치하면 바로 되돌아오면 일부 테스트 집합이 통과할 수 없음을 발견할 수 있다. (나는) 경로를 총계+1한 후에 아래로 계속 계산해야 한다. 만약에 뒤에 있는 데이터가 공교롭게도 0과 같다면...잎 노드에 계산될 때까지.
    /**
     * 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:
        int pathSum(TreeNode* root, int sum) {
            if(root==NULL)
                return 0;
            int cnt=0;
            int target=sum;
            cnt=preOrder(root,cnt,0,target);
            return cnt;
        }
    private:
        int preOrder(TreeNode* root,int cnt,int sum,int target){    // 
            if(root!=NULL){
                cnt=pathCnt(root,cnt,0,target);
                cnt=preOrder(root->left,cnt,0,target);
                cnt=preOrder(root->right,cnt,0,target);
            }
            return cnt;
        }
        int pathCnt(TreeNode* root,int cnt,int sum,int target){
            sum+=root->val;
            if(sum==target){    // 
                ++cnt;
            }
            int temp=sum;   // 
            if(root->left){
                cnt=pathCnt(root->left,cnt,sum,target);
            }
            sum=temp;
            if(root->right){
                 cnt=pathCnt(root->right,cnt,sum,target);
            }
            return cnt;
        }
    };

    좋은 웹페이지 즐겨찾기