【LeetCode】Path Sum(路徑總和)
阿新 • • 發佈:2018-12-01
這道題是LeetCode裡的第112道題。是我在學資料結構——二叉樹的時候碰見的題。
題目要求:
給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。
說明: 葉子節點是指沒有子節點的節點。
示例:
給定如下二叉樹,以及目標和sum = 22
,5 / \ 4 8 / / \ 11 13 4 / \ \ 7 21返回
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); } };