-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathOctomber_9.cpp
30 lines (30 loc) · 884 Bytes
/
Octomber_9.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
// https://leetcode.com/problems/two-sum-iv-input-is-a-bst/
// 653. Two Sum IV - Input is a BST
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
bool dfs(TreeNode* root,unordered_map<int,int>&mp,int k){
if(root == NULL){
return false;
}
if(mp.count(k-root->val)){
return true;
}
mp[root->val]++;
return dfs(root->left,mp,k) || dfs(root->right,mp,k);
}
bool findTarget(TreeNode* root, int k) {
unordered_map<int,int>mp;
return dfs(root,mp,k);
}
};