diff options
| author | crupest <crupest@outlook.com> | 2021-02-23 21:07:19 +0800 |
|---|---|---|
| committer | crupest <crupest@outlook.com> | 2021-02-23 21:07:19 +0800 |
| commit | e353213dfb73259a5d33cc6e9d5cd2373c683be9 (patch) | |
| tree | e879936e7e9c62f9946249cdeb470afeef1c1500 /works/solutions/leetcode/cpp/paths-with-sums-lcci.cpp | |
| parent | 7e3e89a3f0ecc49c33334f423dbde3a5fdc57abd (diff) | |
| download | crupest-e353213dfb73259a5d33cc6e9d5cd2373c683be9.tar.gz crupest-e353213dfb73259a5d33cc6e9d5cd2373c683be9.tar.bz2 crupest-e353213dfb73259a5d33cc6e9d5cd2373c683be9.zip | |
import(solutions): Move leetcode solutions to subdir.
Diffstat (limited to 'works/solutions/leetcode/cpp/paths-with-sums-lcci.cpp')
| -rw-r--r-- | works/solutions/leetcode/cpp/paths-with-sums-lcci.cpp | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/works/solutions/leetcode/cpp/paths-with-sums-lcci.cpp b/works/solutions/leetcode/cpp/paths-with-sums-lcci.cpp new file mode 100644 index 0000000..697182d --- /dev/null +++ b/works/solutions/leetcode/cpp/paths-with-sums-lcci.cpp @@ -0,0 +1,46 @@ +#include <cstddef>
+
+struct TreeNode
+{
+ int val;
+ TreeNode *left;
+ TreeNode *right;
+ TreeNode(int x) : val(x), left(NULL), right(NULL) {}
+};
+
+#include <vector>
+
+class Solution
+{
+public:
+ std::vector<int> current;
+
+ int dfs(TreeNode *root, int sum)
+ {
+ if (root == nullptr)
+ return 0;
+
+ current.push_back(root->val);
+
+ int s = 0;
+ int count = 0;
+
+ for (auto iter = current.crbegin(); iter != current.crend(); ++iter)
+ {
+ s += *iter;
+ if (s == sum)
+ count++;
+ }
+
+ count += dfs(root->left, sum) + dfs(root->right, sum);
+
+ current.pop_back();
+
+ return count;
+ }
+
+ int pathSum(TreeNode *root, int sum)
+ {
+ return dfs(root, sum);
+ }
+};
|