簡體   English   中英

如何去除二叉樹的葉子?

[英]How do I remove the leaves of a binary tree?

我試圖去除所有的葉子。 我知道葉子沒有孩子,這就是我迄今為止所擁有的。

 public void removeLeaves(BinaryTree n){  

    if (n.left == null && n.right == null){

        n = null;

    }

    if (n.left != null)

        removeLeaves(n.left);

    if (n.right != null)

        removeLeaves(n.right);

}

n = null; 不會幫助你,因為n只是你的函數的局部變量。 相反,您需要設置n.left = null; n.right = null; 在父母身上。

我不會給你一個完整的解決方案,因為這聞起來很像家庭作業,但是你可以,例如,向你的函數添加一個返回值來指示有問題的節點是否是葉節點,並在父級(在調用removeLeaves )。

如果你這樣分解它會容易得多:

public void removeLeaves(BinaryTree n){
  if (n.left != null) {
    if (n.left.isLeaf()) {
      n.removeLeftChild();
    } else {
      removeLeaves(n.left);
    }
  }
  // repeat for right child
  // ...
}

isLeafremoveLeftChildremoveRightChild應該很容易實現。

而不是 n = null,它應該是:

if(n.parent != null)
  {
    if(n.parent.left == n)
    {
      n.parent.left = null;
    } 
    else if(n.parent.right == n)
    {
      n.parent.right == null);
    }
  }

由於 Java 通過值傳遞引用n = null; 根本行不通。 這條線 n 指向葉子,現在指向空。 所以你實際上並沒有從父級中刪除它,你只是重新路由一個虛擬的本地引用。 對於解決方案,請按照 Matthew 的建議進行操作。

這是一個從二叉樹中刪除葉節點的簡單java方法

public BinaryTreeNode removeLeafNode(BinaryTreeNode root) {
    if (root == null)
        return null;
    else {
        if (root.getLeft() == null && root.getRight() == null) {     //if both left and right child are null
            root = null;                                             //delete it (by assigning null)
        } else {
            root.setLeft(removeLeafNode(root.getLeft()));            //set new left node 
            root.setRight(removeLeafNode(root.getRight()));          //set new right node   
        }
        return root;
    }

}

使用 recusrion 的簡單方法。

 public static Node removeLeaves(Node root){
          if (root == null) { 
            return null; 
        } 
        if (root.left == null && root.right == null) { 
            return null; 
        } 

        root.left = removeLeaves(root.left); 
        root.right = removeLeaves(root.right); 

        return root;
  }
 /* @author abhineet*/

public class DeleteLeafNodes {


    static class Node{
        int data;
        Node leftNode;
        Node rightNode;
        Node(int value){
            this.data = value;
            this.leftNode = null;
            this.rightNode = null;
        }
    }



    public static void main(String[] args) {

        Node root = new Node(1);
        Node lNode = new Node(2);
        lNode.leftNode = new Node(4);
        root.leftNode = lNode;
        Node rNode = new Node(3);
        rNode.rightNode = new Node(5);
        root.rightNode = rNode;
        printTree(root);
        deleteAllLeafNodes(root, null,0);
        System.out.println("After deleting leaf nodes::");
        printTree(root);

    }

    public static void deleteAllLeafNodes(Node root, Node parent, int direction){
        if(root != null && root.leftNode == null && root.rightNode == null){
            if(direction == 0){
                parent.leftNode = null;
            }else{
                parent.rightNode = null;
            }

        }
        if(root != null && (root.leftNode != null || root.rightNode != null)){
            deleteAllLeafNodes(root.leftNode, root, 0);
            deleteAllLeafNodes(root.rightNode, root, 1);
        }

    }
    public static void printTree(Node root){
        if(root != null){
            System.out.println(root.data);
            printTree(root.leftNode);
            printTree(root.rightNode);
        }
    }

}

這應該有效-

public boolean removeLeaves(Node n){  
    boolean isLeaf = false;
    if (n.left == null && n.right == null){
        return true;
        //n = null;
    }

    if (n!=null && n.left != null){

       isLeaf = removeLeaves(n.left);
       if(isLeaf) n.left=null; //remove left leaf
    }

    if (n!=null && n.right != null){

        isLeaf = removeLeaves(n.right);
        if(b) n.right=null; //remove right leaf
    }
    return false;

}

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM