Đệ quy với các phần tử trả về lợi nhuận trong cây

Question

Tôi có một hàm đệ quy trả về tất cả các nút con, cho nút gốc bắt đầu.

private IEnumerable<Node> getAllNodesRecursively(Node subnode) 
{ 
    foreach (Node node in subnode.Nodes) 
     getAllNodesRecursively(node); 

    yield return subnode; 
} 

Đối với cấu trúc cây sau:

A 
| 
+--B 
| 
+--C 
| | 
| +--D 
| 
+--E 

Khi tôi cố gắng lặp như vậy:

foreach (Node n in getAllNodesRecursively(a)) 
{ 
    Console.WriteLine(n); 
} 

hàm trả về chỉ A giá trị.

Tôi muốn sử dụng trả lại lợi nhuận với đệ quy và truy xuất các phần tử trong Đơn đặt hàng trước (A, B, C, D, E trong ví dụ này).

(Nếu tôi đặt lợi nhuận về lợi nhuận trước khi foreach, thì việc cho phép sẽ không bao giờ xảy ra).

Điều này có khả thi không?

Answer 1

Các bạn đã thử một cái gì đó như:

private IEnumerable<Node> getAllNodesRecursively(Node subnode) 
{ 
    // Return the parent before its children 
    yield return subnode; 

    foreach (Node node in subnode.Nodes) 
    { 
     foreach(Node n in getAllNodesRecursively(node)) 
     { 
      yield return n; 
     } 
    } 
} 

triển khai của bạn đang kêu gọi getAllNodesRecursively đệ quy, nhưng bỏ qua giá trị trả về của nó.

Answer 2

Có thể, chỉ cần đặt yield return trước foreach. Bạn đang suy nghĩ về hành vi của câu lệnh bình thường return.

Answer 3

 public IEnumerable<int> preOrder(Node root) 
     { 
      if (root == null) 
       yield break; 

      yield return root.val; 

      if (root.left != null) 
       foreach (int i in preOrder(root.left)) 
        yield return i; 

      if (root.right != null) 
       foreach (int i in preOrder(root.right)) 
        yield return i; 
     }