劍指 Offer II 056. 二叉搜索樹中兩個節點之和

Ideas

這題有點類似一個組合題，首先是通過二叉樹遍歷得到一個序列，然后再通過LeetCode Algorithm 1. 兩數之和的方法查找是否存在兩數之和就OK了。

二叉樹的遍歷直接默寫模板，然后查找兩數之和可以通過哈希表實現。

Code

C++

class Solution { public:vector<int> preorder_traversal(TreeNode* node) {vector<int> ans;stack<TreeNode*> stk;stk.push(node);while (!stk.empty()) {TreeNode* item = stk.top();stk.pop();ans.push_back(item->val);if (item->right != nullptr) {stk.push(item->right);}if (item->left != nullptr) {stk.push(item->left);}}return ans;}bool findTarget(TreeNode* root, int k) {if (!root) {return false;}vector<int> preorder_list = preorder_traversal(root);unordered_map<int, int> hash_table;for (int i = 0; i < preorder_list.size(); i++) {auto flag = hash_table.find(k - preorder_list[i]);if (flag != hash_table.end()) {return true;}hash_table[preorder_list[i]] = i;}return false;} };

總結

以上是生活随笔為你收集整理的LeetCode Algorithm 剑指 Offer II 056. 二叉搜索树中两个节点之和的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。