[C언어 LeetCode] 113.경로 총 II (M)

1954 단어 LeetCode

두 갈래 나무와 목표와 뿌리 노드에서 잎 노드까지의 모든 경로를 찾는 것은 목표와 같은 경로입니다.
설명: 잎 노드는 하위 노드가 없는 노드를 가리킨다.
예: 다음과 같은 두 갈래 트리와 목표와sum=22,
              5              /\            4   8            / /\          11  13  4          / \   /\        7    2  5   1
반환:
[    [5,4,11,2],    [5,8,4,5] ]
출처: 리코드(LeetCode) 링크:https://leetcode-cn.com/problems/path-sum-ii저작권은 인터넷 소유에 귀속된다.상업 전재는 정부에 연락하여 권한을 부여하고, 비상업 전재는 출처를 명시해 주십시오.

112문제의 업그레이드판입니다. 이 두 갈래 트리 문제는 경로를 기록해야 하고temp수조가 필요합니다.이 문제는 좀 틀리기 쉽다.전삼이 너무 많아서
void preorder(struct TreeNode* root, int sum, int *size, int* colsize, int **result, int *temp, int now) {
    int i;
    
    if(root == NULL) return;
    
    if((root->left==NULL) && (root->right == NULL) && (sum == root->val)) {
        temp[now++] = root->val;

        colsize[*size] = now;
        result[*size] = (int *)malloc(sizeof(int) * now);
        
        for(i = 0; i < now; i++) {
            result[*size][i] = temp[i];
        }
        
        (*size)++;
    }
        
    temp[now++] = root->val;
    sum -= root->val;
        
    preorder(root->left, sum, size, colsize, result, temp, now);
    preorder(root->right, sum, size, colsize, result, temp, now); 
}

/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int** pathSum(struct TreeNode* root, int sum, int* returnSize, int** returnColumnSizes){
    int **result;
    int size = 0;
    int *colsize;
    int temp[1024] = {0};
    
    result = (int **)malloc(sizeof(int *) * 1024);
    colsize = (int *)malloc(sizeof(int *) * 1024);
    
    preorder(root, sum, &size, colsize, result, temp, 0);
    
    *returnSize = size;
    *returnColumnSizes = colsize;
    
    return result;
    
}

좋은 웹페이지 즐겨찾기