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 | 951ff37ae58fc27d178b1650a8b2af6b01c960f1 (patch) | |
| tree | 8ab88769b4d5e5764602cbdd2724757d4473706a | |
| parent | 78e272d454c3cde15a48ca59e5825aaba1b2727d (diff) | |
| download | crupest-951ff37ae58fc27d178b1650a8b2af6b01c960f1.tar.gz crupest-951ff37ae58fc27d178b1650a8b2af6b01c960f1.tar.bz2 crupest-951ff37ae58fc27d178b1650a8b2af6b01c960f1.zip | |
import(solutions): Add problem paths-with-sums-lcci.
| -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);
+ }
+};
|