diff options
| author | crupest <crupest@outlook.com> | 2020-10-02 22:23:55 +0800 |
|---|---|---|
| committer | crupest <crupest@outlook.com> | 2020-10-02 22:23:55 +0800 |
| commit | b08eced5352c9118b539b8a4e6dc1bd1f4c7f371 (patch) | |
| tree | 0338827fc8f48d92c1a531f81e3f9e9fc0375d07 /works | |
| parent | a62676feb90dadca5d0cac9a9046f9a6641cc6a2 (diff) | |
| download | crupest-b08eced5352c9118b539b8a4e6dc1bd1f4c7f371.tar.gz crupest-b08eced5352c9118b539b8a4e6dc1bd1f4c7f371.tar.bz2 crupest-b08eced5352c9118b539b8a4e6dc1bd1f4c7f371.zip | |
import(solutions): Add problem paths-with-sums-lcci.
Diffstat (limited to 'works')
| -rw-r--r-- | works/solutions/cpp/paths-with-sums-lcci.cpp | 46 |
1 files changed, 46 insertions, 0 deletions
diff --git a/works/solutions/cpp/paths-with-sums-lcci.cpp b/works/solutions/cpp/paths-with-sums-lcci.cpp new file mode 100644 index 0000000..697182d --- /dev/null +++ b/works/solutions/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);
+ }
+};
|