|
55 | 55 |
|
56 | 56 | <!-- 这里可写通用的实现逻辑 -->
|
57 | 57 |
|
| 58 | +先预处理父子节点的关系,然后 DFS 搜索即可。 |
| 59 | + |
58 | 60 | <!-- tabs:start -->
|
59 | 61 |
|
60 | 62 | ### **Python3**
|
@@ -177,6 +179,71 @@ class Solution {
|
177 | 179 | }
|
178 | 180 | ```
|
179 | 181 |
|
| 182 | +### **C++** |
| 183 | + |
| 184 | +```cpp |
| 185 | +/** |
| 186 | + * Definition for a binary tree node. |
| 187 | + * struct TreeNode { |
| 188 | + * int val; |
| 189 | + * TreeNode *left; |
| 190 | + * TreeNode *right; |
| 191 | + * TreeNode() : val(0), left(nullptr), right(nullptr) {} |
| 192 | + * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} |
| 193 | + * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} |
| 194 | + * }; |
| 195 | + */ |
| 196 | +class Solution { |
| 197 | +public: |
| 198 | + unordered_map<int, vector<pair<int, char>>> edges; |
| 199 | + unordered_set<int> visited; |
| 200 | + string ans; |
| 201 | + |
| 202 | + string getDirections(TreeNode* root, int startValue, int destValue) { |
| 203 | + ans = ""; |
| 204 | + traverse(root); |
| 205 | + string t = ""; |
| 206 | + dfs(startValue, destValue, t); |
| 207 | + return ans; |
| 208 | + } |
| 209 | + |
| 210 | + void traverse(TreeNode* root) { |
| 211 | + if (!root) return; |
| 212 | + if (root->left) |
| 213 | + { |
| 214 | + edges[root->val].push_back({root->left->val, 'L'}); |
| 215 | + edges[root->left->val].push_back({root->val, 'U'}); |
| 216 | + } |
| 217 | + if (root->right) |
| 218 | + { |
| 219 | + edges[root->val].push_back({root->right->val, 'R'}); |
| 220 | + edges[root->right->val].push_back({root->val, 'U'}); |
| 221 | + } |
| 222 | + traverse(root->left); |
| 223 | + traverse(root->right); |
| 224 | + } |
| 225 | + |
| 226 | + void dfs(int start, int dest, string& t) { |
| 227 | + if (visited.count(start)) return; |
| 228 | + if (start == dest) |
| 229 | + { |
| 230 | + if (ans == "" || ans.size() > t.size()) ans = t; |
| 231 | + return; |
| 232 | + } |
| 233 | + visited.insert(start); |
| 234 | + if (edges.count(start)) |
| 235 | + { |
| 236 | + for (auto& item : edges[start]) |
| 237 | + { |
| 238 | + t += item.second; |
| 239 | + dfs(item.first, dest, t); |
| 240 | + t.pop_back(); |
| 241 | + } |
| 242 | + } |
| 243 | + } |
| 244 | +}; |
| 245 | +``` |
| 246 | +
|
180 | 247 | ### **TypeScript**
|
181 | 248 |
|
182 | 249 | <!-- 这里可写当前语言的特殊实现逻辑 -->
|
|
0 commit comments