1. 程式人生 > >【LeetCode】Path Sum(路徑總和)

【LeetCode】Path Sum(路徑總和)

這道題是LeetCode裡的第112道題。是我在學資料結構——二叉樹的時候碰見的題。

題目要求:

給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。

說明: 葉子節點是指沒有子節點的節點。

示例: 
給定如下二叉樹,以及目標和 sum = 22

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2
1

返回 true, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2

本來是一道簡單的題目,卻因為開頭的思路不正確,或者說是沒有抓住重點,認真審題,導致提交過程中屢次碰壁。

下面給出我之前的思路:

起初是為了方便使用遞迴,然後遞迴引數使用當前節點的左右子節點,和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 {
public:
    bool isroot=1;//這是拿來標記根節點的
    bool hasPathSum(TreeNode* root, int sum) {
        if (isroot) {
            isroot = 0;
            if(root == NULL) return false;
            if(root->left == NULL && root->right == NULL && sum == root->val) return true;
            else return false;
        }
        if (root == NULL && sum == 0) return true;
        if (root == NULL && sum != 0) return false;
        return hasPathSum(root->left, sum - root->val) ||
            hasPathSum(root->right, sum - root->val);
    }
};

修改多次,每次上傳都無法通過。然後沒辦法調到自己的編譯器中除錯,最後發現了問題,最大的問題就是題目要求的是從根節點到葉子節點的路徑,而我這個演算法是不能保證最後的節點是否是葉子節點。

發現了問題自然需要解決問題,然後我重寫了一個演算法,反而還更簡短了:

/**
 * 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:
    bool hasPathSum(TreeNode* root, int sum) {
        if (root == NULL)
            return false;
        sum = sum - root->val;
        if (root->left == NULL && root->right == NULL && sum == 0)
            return true;
        return hasPathSum(root->left, sum) ||
            hasPathSum(root->right, sum);
    }
};

貼個結果: