-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathmain.cpp
58 lines (52 loc) · 1.93 KB
/
main.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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
/**
* 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:
TreeNode *createBinaryTree(vector<vector<int>> &descriptions)
{
unordered_map<int, TreeNode *> mp; // Lecture des valeurs de la description
unordered_set<int> set; // Set pour identifier les potentiels racines
// Parcours de chaque description
for (vector<int> &v : descriptions)
{
TreeNode *parent = mp[v[0]]; // Récupère le noeud parent depuis la map
TreeNode *child = mp[v[1]]; // Récupère le noeud enfant depuis la map
// Si le noeud enfant n'existe pas encore
if (!child)
{
child = new TreeNode(v[1]); // Crée un nouveau noeud enfant
mp[v[1]] = child; // Ajoute le noeud enfant à la map
}
// Si le noeud parent n'existe pas encore
if (!parent)
{
parent = new TreeNode(v[0]); // Crée un nouveau noeud parent
mp[v[0]] = parent; // Ajoute le noeud parent à la map
set.insert(v[0]); // Ajoute le parent au set des potentiels racines
}
// Affecter l'enfant au parent
if (v[2])
{
parent->left = child;
}
else
{
parent->right = child;
}
// L'enfant ne peut pas être une racine, donc on l'enlève du set
set.erase(v[1]);
}
// Le seul élément restant dans le set est la racine de l'arbre
return mp[*set.begin()];
}
};