forked from ttzztztz/leetcodeAlgorithms
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path1624. Max Distance.cpp
More file actions
44 lines (38 loc) · 1.14 KB
/
1624. Max Distance.cpp
File metadata and controls
44 lines (38 loc) · 1.14 KB
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
class Solution {
public:
class TreeNode {
public:
bool flag;
TreeNode* left, *right;
TreeNode() : flag(false), left(nullptr), right(nullptr) {};
};
int getAns(vector<string> &str) {
root = new TreeNode();
for (const string& s : str) insert(s);
dfs(root);
return answer;
}
private:
TreeNode* root;
int answer = 0;
void insert(const string& s) {
TreeNode* cur = root;
for (const char ch : s) {
TreeNode** target = nullptr;
if (ch == '0') target = &(cur->left);
else target = &(cur->right);
if (*target == nullptr) *target = new TreeNode();
cur = *target;
}
cur->flag = true;
}
int dfs(TreeNode* u) {
if (u == nullptr) return 0;
int left = dfs(u->left);
int right = dfs(u->right);
if (left && right) answer = max(answer, left + right);
if (u->flag && left) answer = max(answer, left);
if (u->flag && right) answer = max(answer, right);
return max(left, right) + 1;
}
};