【leetcode】222（Medium）Count Complete Tree Nodes

原创于 2019-05-06 19:57:58 发布 · 179 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

257 篇文章

订阅专栏

博客介绍了二叉树遍历的解题思路，可采用morris遍历或递归方法，并分别给出了morris和递归的提交代码及运行结果，聚焦于信息技术领域的算法解题。

解题思路

可以用morris遍历，或者递归

提交代码：morris

class Solution {
    public int countNodes(TreeNode root) {
        int res=0;
        
        TreeNode p1=root,p2=root;
        while(p1!=null) {
        	if(p1.left!=null) {
        		p2=p1.left;
        		while(p2.right!=null&&p2.right!=p1)
        			p2=p2.right;
        		
        		if(p2.right==null) {
        			p2.right=p1;
        			p1=p1.left;
        		}
        		else {
        			res++;
        			p2.right=null;
        			p1=p1.right;
        		}
        	}else {
        		res++;
        		p1=p1.right;
        	}
        }
        return res;
    }
}

运行结果

在这里插入图片描述

提交代码：递归

class Solution {
	public int countNodes(TreeNode root) {
		if(root==null)	return 0;
		return countNodes(root.left)+countNodes(root.right)+1;
	}
}

运行结果

在这里插入图片描述