一、問題描述

1、只記錄和

給定一棵二叉樹和一個和，確定樹是否有從根到葉的路徑，使得沿路徑加起來的所有值等於給定的總和。
【示例】
給定下面的二叉樹 以及待求和 sum = 22,
          5
        /   \
      4     8
     /     /  \
   11  13   4
   /  \     \
 7    2      1

返回 true, 因為存在一個從根到葉的路徑：5->4->11->2 和為：22.

2、記錄路徑

給定一個二叉樹和一個和，找到每個路徑的總和等於給定總和的所有根到葉路徑。

【舉例】給定下面的二叉樹，以及和 sum = 22,
            5
         /     \
       4       8
      /       /   \
   11     13    4
   /  \    /  \
 7    2  5   1
返回:
[
   [5,4,11,2],
   [5,8,4,5]

]

二、解題思路

1、只記錄和

幾個注意點：
    1）題目要求返回true/false，不需要記錄路徑
    2）給定結點沒說是正整數，因此必須走到葉子結點才知道是否找到，沒法剪枝，只能樸素深度搜索

    3）一旦找到sum結果，立即return

2、記錄路徑

dfs

三、解題演算法

1、只記錄和

/*********************************************
Author:tmw
date:2018-5-8
*********************************************/
#include <stdio.h>
#include <stdlib.h>

typedef struct TreeNode
{
    int val;
    struct TreeNode* left;
    struct TreeNode* right;
}TreeNode;

bool hasPathSum(TreeNode* root, int sum)
{
    if( !root ) return false;

    /**走到葉子結點，判斷當前sum值與葉子結點值是否相等**/
    if( root->left == NULL && root->right == NULL )
        return root->val == sum;
    /**沒走到葉子結點：左右孩子都可以走，更新sum**/
    return hasPathSum(root->left,sum-root->val) || hasPathSum(root->right,sum-root->val);
}
 

2、記錄路徑

/*******************************************
Author:tmw
date:2018-5-8
*******************************************/
#include <stdio.h>
#include <stdlib.h>

typedef struct TreeNode
{
    int val;
    struct TreeNode* left;
    struct TreeNode* right;
}TreeNode;

/**
 *returnSize  -- 返回的結果陣列有多少組
 **columnSizes -- 返回的結果陣列每一組有多少個元素
*/
#define MAX_SIZE 1000
/**
    *cur_array ： 用於暫存中間值
    **result  : 最終滿足和為sum的所有路徑
    *colmn_size : 用於記錄每條路徑有多少元素
**/
int i=0;
int count = 0;
int colmn_size[MAX_SIZE];
void get_pathSum( TreeNode* root, int sum, int* cur_array, int** result )
{
    if( root==NULL ) return;

    cur_array[i++] = root->val;

    /**當遍歷到葉子結點時，判斷該路徑下的和值是否滿足sum，滿足則記錄到result陣列中**/
    if( root->left == NULL && root->right == NULL )
    {
        /**滿足則記錄到result陣列中**/
        if( root->val == sum )
        {
            int j;
            for( j=0; j<i; j++ )
                result[count][j] = cur_array[j];
            colmn_size[count] = i;
            count++;
        }
    }

    /**記錄左子樹路徑**/
    get_pathSum( root->left, sum-root->val, cur_array, result);
    /**記錄右子樹路徑**/
    get_pathSum( root->right, sum-root->val, cur_array, result);
}

int** pathSum(TreeNode* root, int sum, int** columnSizes, int* returnSize)
{
    if( root == NULL ) return NULL;

    /**自定義二維的結果陣列，用來儲存最終結果並返回**/
    int** result = (int**)malloc(MAX_SIZE*sizeof(int*));
    int ii;
    for( ii=0; ii<MAX_SIZE; ii++ )
        result[ii] = (int*)malloc(MAX_SIZE*sizeof(int));

    /**自定義臨時一維陣列用於暫存結果值**/
    int* temp_res = (int*)malloc(MAX_SIZE*sizeof(int));

    get_pathSum(root,sum,temp_res,result);

    /**賦值**/
    *returnSize = count;
    printf("%d",count);
    for( ii=0;ii<count;ii++ )
        (*columnSizes)[ii] = colmn_size[ii];

    return result;
}
 

夢想還是要有的，萬一實現了呢~~~ヾ(◍°∇°◍)ﾉﾞ~~~~