Подсчёт листьев бинарного дерева на Java — рекурсивный алгоритм

Лист — это узел дерева, у которого нет потомков (и левый, и правый указатели равны null). Чтобы посчитать количество листьев, удобно использовать рекурсию: если узел — лист, возвращаем 1, иначе суммируем количество листьев в левом и правом поддереве.

Чтобы понять этот пример, нужно знать:

• Классы и объекты в Java

• Методы в Java

Пример: подсчёт листьев дерева на Java

class Node {
  int item;
  Node left, right;

  public Node(int key) {
  item = key;
  left = right = null;
  }
}

class Main {
  // root of Tree
  Node root;

  Main() {
  root = null;
  }

  // method to count leaf nodes
  public static int countLeaf(Node node) {
    if(node == null) {
      return 0;
    }

    // if left and right of the node is null
    // it is leaf node
    if (node.left == null && node.right == null) {
      return 1;
    }
    else {
      return countLeaf(node.left) + countLeaf(node.right);
    }
  }

  public static void main(String[] args) {

    // create an object of Tree
    Main tree = new Main();

    // create nodes of tree
    tree.root = new Node(5);
    tree.root.left = new Node(3);
    tree.root.right = new Node(8);

    // create child nodes of left child
    tree.root.left.left = new Node(2);
    tree.root.left.right = new Node(4);

    // create child nodes of right child
    tree.root.right.left = new Node(7);
    tree.root.right.right = new Node(9);

    // call method to count leaf nodes
    int leafNodes = countLeaf(tree.root);
    System.out.println("Total Leaf Nodes = " + leafNodes);
  }
}

Вывод:

Total Leaf Nodes = 4

В этом примере мы построили дерево из 7 узлов: корень 5, его потомки 3 и 8, и по два потомка у каждого из них. Листьями являются узлы 2, 4, 7 и 9 — всего 4 листа, что и вывела программа.

Примечание

Для подсчёта листьев мы использовали рекурсию: метод countLeaf() вызывает сам себя для левого и правого поддерева.