add: 220416 [cpp]
This commit is contained in:
parent
93216ec220
commit
e5549fb0dd
|
|
@ -1,26 +1,26 @@
|
||||||
#include <functional>
|
#include <functional>
|
||||||
|
|
||||||
struct TreeNode {
|
struct TreeNode {
|
||||||
int val;
|
int val;
|
||||||
TreeNode* left;
|
TreeNode* left;
|
||||||
TreeNode* right;
|
TreeNode* right;
|
||||||
explicit TreeNode(int x, TreeNode* left = nullptr, TreeNode* right = nullptr) : val(x), left(left), right(right) {}
|
explicit TreeNode(int x, TreeNode* left = nullptr, TreeNode* right = nullptr) : val(x), left(left), right(right) {}
|
||||||
};
|
};
|
||||||
|
|
||||||
class Solution {
|
class Solution {
|
||||||
public:
|
public:
|
||||||
static TreeNode* convertBST(TreeNode* root) {
|
static TreeNode* convertBST(TreeNode* root) {
|
||||||
if (!root)
|
if (!root)
|
||||||
return nullptr;
|
return nullptr;
|
||||||
int x = 0;
|
int x = 0;
|
||||||
std::function<void(TreeNode*)> reversedInOrderTraverse = [&](TreeNode* r) {
|
std::function<void(TreeNode*)> reversedInOrderTraverse = [&](TreeNode* r) {
|
||||||
if (r->right)
|
if (r->right)
|
||||||
reversedInOrderTraverse(r->right);
|
reversedInOrderTraverse(r->right);
|
||||||
r->val = (x += r->val);
|
r->val = (x += r->val);
|
||||||
if (r->left)
|
if (r->left)
|
||||||
reversedInOrderTraverse(r->left);
|
reversedInOrderTraverse(r->left);
|
||||||
};
|
};
|
||||||
reversedInOrderTraverse(root);
|
reversedInOrderTraverse(root);
|
||||||
return root;
|
return root;
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue