二叉树相关练习

未完待续

/**
 * 二叉树节点类
 * @author Administrator
 *

 */
public class BinaryTreeNode {
    public Integer data; // data
    public BinaryTreeNode leftNode; // 左孩子
    public BinaryTreeNode rightNode; // 右孩子
}
package com.binarytree;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import java.util.Stack;
/** 
 *  
 * TODO: 一定要能熟练地写出所有问题的递归和非递归做法! 
 * 
 * 1. 求二叉树中的节点个数: getNodeNumRec(递归),getNodeNum(迭代) 
 * 2. 求二叉树的深度: getDepthRec(递归),getDepth  
 * 3. 前序遍历,中序遍历,后序遍历: preorderTraversalRec, preorderTraversal, inorderTraversalRec, postorderTraversalRec 
 * (https://en.wikipedia.org/wiki/Tree_traversal#Pre-order_2) 
 * 4.分层遍历二叉树(按层次从上往下,从左往右): levelTraversal, levelTraversalRec(递归解法!) 
 * 5. 将二叉查找树变为有序的双向链表: convertBST2DLLRec, convertBST2DLL 
 * 6. 求二叉树第K层的节点个数:getNodeNumKthLevelRec, getNodeNumKthLevel 
 * 7. 求二叉树中叶子节点的个数:getNodeNumLeafRec, getNodeNumLeaf 
 * 8. 判断两棵二叉树是否相同的树:isSameRec, isSame 
 * 9. 判断二叉树是不是平衡二叉树:isAVLRec 
 * 10. 求二叉树的镜像(破坏和不破坏原来的树两种情况):mirrorRec, mirrorCopyRec 
 * 10.1 判断两个树是否互相镜像:isMirrorRec 
 * 11. 求二叉树中两个节点的最低公共祖先节点:getLastCommonParent, getLastCommonParentRec, getLastCommonParentRec2 
 * 12. 求二叉树中节点的最大距离:getMaxDistanceRec 
 * 13. 由前序遍历序列和中序遍历序列重建二叉树:rebuildBinaryTreeRec 
 * 14.判断二叉树是不是完全二叉树:isCompleteBinaryTree, isCompleteBinaryTreeRec 
 *  
 */  
public class BinaryTreeSummary {

    /**
     * 递归前序遍历
     * @param root
     */
    public static void preOrderRec(BinaryTreeNode root){
        if(root != null) {
            System.out.print(root.data+" ");
            preOrderRec(root.leftNode);
            preOrderRec(root.rightNode);
        }
    }
    
    /**
     * 非递归前序遍历
     * @param root
     */
    public static void preOrder(BinaryTreeNode root){
        Stack<BinaryTreeNode> s = new Stack<BinaryTreeNode>();
        BinaryTreeNode p = root;
        while(p!=null || s.size()>0) {
            if(p!=null) {
                System.out.print(p.data+" ");
                s.push(p);
                p = p.leftNode;
            }
            else {
                p = s.pop();
                p = p.rightNode;
            }
        }
    }
    
    /**
     * 非递归中序遍历
     * @param root
     */
    public static void inOrder(BinaryTreeNode root) {
        Stack<BinaryTreeNode> s = new Stack<BinaryTreeNode>();
        BinaryTreeNode p = root;
        while(p!=null || s.size()>0) {
            if(p!=null) {
                s.push(p);
                p = p.leftNode;
            }
            else {
                p = s.pop();
                System.out.print(p.data+" ");
                p = p.rightNode;
            }
        }
    }
    
    /**
     * 递归中序遍历
     * @param root
     */
    public static void inOrderRec(BinaryTreeNode root) {
        if(root != null) {
            inOrderRec(root.leftNode);
            System.out.print(root.data+" ");
            inOrderRec(root.rightNode);
        }
    }
    
    /**
     * 非递归层次遍历二叉树
     * 队列实现
     * @param root
     */
    public static void printTreeFromTopToBottom(BinaryTreeNode root) {
        Queue<BinaryTreeNode> q = new LinkedList<BinaryTreeNode>();
        q.add(root);
        while(q.size() > 0) {
            BinaryTreeNode node = q.poll();
            System.out.print(node.data+" ");
            if(node.leftNode != null) {
                q.add(node.leftNode);
            }
            if(node.rightNode != null) {
                q.add(node.rightNode);
            }
        }
    }
    
    /**
     * 递归层次遍历二叉树
     * @param root
     */
    public static void printTreeFromTopToBottomRec(BinaryTreeNode root) {
        ArrayList<ArrayList<Integer>> ret = new ArrayList<ArrayList<Integer>>();  
        dfs(root, 0, ret);  
        System.out.println(ret);  
    }  
      
    private static void dfs(BinaryTreeNode root, int level, ArrayList<ArrayList<Integer>> ret){  
        if(root == null){  
            return;  
        }  
          
        // 添加一个新的ArrayList表示新的一层  
        if(level >= ret.size()){  
            ret.add(new ArrayList<Integer>());  
        }  
          
        ret.get(level).add(root.data);   // 把节点添加到表示那一层的ArrayList里  
        dfs(root.leftNode, level+1, ret);       // 递归处理下一层的左子树和右子树  
        dfs(root.rightNode, level+1, ret);  
    }
    
    
    /**
     * 重建二叉树
     * @param preOrder
     * @param inOrder
     * @return
     */
    public static BinaryTreeNode rebuildBinaryTree(List<Integer>  preOrder, List<Integer> inOrder ) {
        
        BinaryTreeNode root = null;  
        List<Integer> leftPreOrder;  
        List<Integer> rightPreOrder;  
        List<Integer> leftInorder;  
        List<Integer> rightInorder;  
        int inorderPos;  
        int preorderPos;  
   
        if ((preOrder.size() != 0) && (inOrder.size() != 0))  
        {  
            // 把preorder的第一个元素作为root  
            root = new BinaryTreeNode();
            root.data = preOrder.get(0);
   
            //  Based upon the current node data seperate the traversals into leftPreorder, rightPreorder,  
            //  leftInorder, rightInorder lists  
            // 因为知道root节点了,所以根据root节点位置,把preorder,inorder分别划分为 root左侧 和 右侧 的两个子区间  
            inorderPos = inOrder.indexOf(preOrder.get(0));      // inorder序列的分割点  
            leftInorder = inOrder.subList(0, inorderPos);  
            rightInorder = inOrder.subList(inorderPos + 1, inOrder.size());  
   
            preorderPos = leftInorder.size();                           // preorder序列的分割点  
            leftPreOrder = preOrder.subList(1, preorderPos + 1);  
            rightPreOrder = preOrder.subList(preorderPos + 1, preOrder.size());  
   
            root.leftNode = BinaryTreeSummary.rebuildBinaryTree(leftPreOrder, leftInorder);        // root的左子树就是preorder和inorder的左侧区间而形成的树  
            root.rightNode = BinaryTreeSummary.rebuildBinaryTree(rightPreOrder, rightInorder); // root的右子树就是preorder和inorder的右侧区间而形成的树  
        }  
        return root;  
    }
    
    /**
     * 二叉树的镜像
     * @param root
     */
    public static void MirrorBinaryTree(BinaryTreeNode root) {
        if(root != null) {
            if(root.leftNode!=null || root.rightNode!=null) {
                BinaryTreeNode x = new BinaryTreeNode();
                x = root.leftNode;
                root.leftNode = root.rightNode;
                root.rightNode = x;
                MirrorBinaryTree(root.rightNode);
                MirrorBinaryTree(root.leftNode);
            }
        }
        
    }
    
    /**
     * 递归求二叉树的节点个数
     * @param root
     * @return
     */
    public static int getNodeNumRec(BinaryTreeNode root) {
        if(root == null) {
            return 0;
        }
        return getNodeNumRec(root.leftNode) + getNodeNumRec(root.rightNode) + 1;
    }
    
    /**
     * 非递归求二叉树的节点个数
     * 通过队列进行层次遍历
     * @param root
     * @return
     */
    public static int getNodeNum(BinaryTreeNode root) {
        int num=0;
        Queue<BinaryTreeNode> q = new LinkedList<BinaryTreeNode>();
        q.add(root);
        while(q.size() > 0) {
            BinaryTreeNode node = q.poll();
            num++;
            if(node.leftNode != null) {
                q.add(node.leftNode);
            }
            if(node.rightNode != null) {
                q.add(node.rightNode);
            }
        }
        return num;
    }
    
    /**
     * 递归求二叉树的深度
     * @param root
     * @return
     */
    public static int getDepthRec(BinaryTreeNode root) {
        if(root == null ) {
            return 0;
        }
        return Math.max(getDepthRec(root.leftNode)+1, getDepthRec(root.rightNode)+1);
    }
    
    /**
     * 非递归求二叉树的深度
     * 队列层次遍历,遍历一层depth++
     * @param root
     * @return
     */
    public static int getDepth(BinaryTreeNode root) {
        int depth=0;
        int currentLevelNodeNums = 1;
        int nextLevelNodeNums = 0;
        Queue<BinaryTreeNode> q = new LinkedList<BinaryTreeNode>();
        if(root == null ) {
            return 0;
        }
        q.add(root);
        while(q.size() > 0) {
            BinaryTreeNode node = q.poll(); // 去除一个节点
            currentLevelNodeNums--; // 当前层数节点减少一个
            
            if(node.leftNode != null) {
                q.add(node.leftNode);
                nextLevelNodeNums++;
            }
            if(node.rightNode != null) {
                q.add(node.rightNode);
                nextLevelNodeNums++;
            }
            if(currentLevelNodeNums == 0) { // 当前层节点数为0  当前层遍历完成
                depth++;
                currentLevelNodeNums = nextLevelNodeNums;
                nextLevelNodeNums = 0;
            }
        }
        return depth;
    }
    
    /**
     * 求二叉树第k层的节点个数
     * @param root
     * @param k
     * @return
     */
    public static int getNodeNumKthLevel(BinaryTreeNode root, int k) {
        int currentLevel = 1;
        int nextLevelNodesNum = 0;
        int currentNodesNum = 1;
        Queue<BinaryTreeNode> s = new LinkedList<BinaryTreeNode>();
        if(root == null) {
            return 0;
        }
        s.add(root);
        while( s.size()>0 && currentLevel<k) {
            BinaryTreeNode node = s.poll();
            currentNodesNum--;
            
            if(node.leftNode!=null) {
                s.add(node.leftNode);
                nextLevelNodesNum++;
            }
            if(node.rightNode!=null) {
                s.add(node.rightNode);
                nextLevelNodesNum++;
            }
            if(currentNodesNum == 0) {
                currentLevel++; // 进入下一层
                currentNodesNum = nextLevelNodesNum;
                nextLevelNodesNum = 0;
            }
            
        }
        return currentNodesNum;
    }
    
    public static void main(String[] args) {
        /*       3
         *    2     5
         *  1   * 4   6
         */
        BinaryTreeNode x = new BinaryTreeNode();
        x.data = 3;
        x.leftNode = new BinaryTreeNode();
        x.leftNode.data = 2;
        x.rightNode = new BinaryTreeNode();
        x.rightNode.data = 5;
        x.rightNode.leftNode = new BinaryTreeNode();
        x.rightNode.leftNode.data = 1;
        x.rightNode.rightNode = new BinaryTreeNode();
        x.rightNode.rightNode.data = 6;
        x.leftNode.leftNode = new BinaryTreeNode();
        x.leftNode.leftNode.data = 4;
//      BinaryTreeSummary.MirrorBinaryTree(x);
//      System.out.println();
//      System.out.println(x.toString());
//      BinaryTreeSummary.printTreeFromTopToBottom(x);
//      System.out.println("asd");
        System.out.println("Start");
        System.out.println(BinaryTreeSummary.getNodeNumKthLevel(x, 0));
        System.out.println("End");
    }
//      List<Integer> x = new ArrayList<Integer>();
//      List<Integer> y = new ArrayList<Integer>();
//      x.add(1);x.add(2);x.add(4);x.add(7);x.add(3);x.add(5);x.add(6);x.add(8);
//      y.add(4);y.add(7);y.add(2);y.add(1);y.add(5);y.add(3);y.add(8);y.add(6);
//      BinaryTreeNode z = BinaryTreeSummary.rebuildBinaryTree(x,y);
//      System.out.println(z.toString());
//      System.out.println();
//  }
}

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,194评论 6 490
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,058评论 2 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 156,780评论 0 346
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,388评论 1 283
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,430评论 5 384
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,764评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,907评论 3 406
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,679评论 0 266
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,122评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,459评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,605评论 1 340
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,270评论 4 329
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,867评论 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,734评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,961评论 1 265
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,297评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,472评论 2 348

推荐阅读更多精彩内容