Morris中序遍历解决LeetCode501

Morris中序遍历

由于在遍历的时候,我们需要记住某种遍历次序的的后驱或者前驱结点,
常见的递归和非递归都是采用栈的方式完成这个过程,有没有内部空间来记录这些后驱或者前驱结点呢?
有,那就是叶结点的左,右孩子结点,因为叶结点的两个孩子结点都是空指针,
如果利用好这些空间,我们就可以在O(1) 的空间完成遍历。

基本思想

Morris 中序遍历过程如下:

1.当前结点的左孩子是否为空,若是则输出当前结点,更当前结点为当前结点的右孩子;否则进入2;

2.在当前结点的左子树中寻找中序遍历下的前驱结点(左子树中最右结点)

	a. 若前驱结点的右孩子为空,则将前驱结点的右孩子指向当前结点,当前结点更新为当前结点的左孩子;进入3;

	b. 若前驱结点的右孩子为当前结点(不为空),将前驱结点的右孩子置NULL,输出当前结点,当前结点更新为当前结点的右孩子,进入3;

3.若当前结点不为空,进入1;否则程序结束;

在这里插入图片描述

  public static void inOrder(TreeNode root) {
        TreeNode cur = root, pre = null;
        for (; cur != null;) {
            if (cur.left != null) {
                pre = cur.left;
                // find predecessor
                while (pre.right != null && pre.right != cur)
                    pre = pre.right;
                if (pre.right == null) {// create thread
                    pre.right = cur;
                    cur = cur.left;
                } else {
                    print(cur);
                    pre.right = null;
                    cur = cur.right;
                }
            } else {
                print(cur);
                cur = cur.right;
            }
        }
    }

那么我们如何解决LeetCode501呢?

我们要明确:我们需要空间复杂度为常数,所以我们才使用Mirros遍历,那么我们需要进行那些操作呢。

我们可以顺序扫描中序遍历序列,用base 记录当前的数字,用count 记录当前数字重复的次数,
用 maxCount 来维护已经扫描过的数当中出现最多的那个数字的出现次数,
用 answer 数组记录出现的众数。每次扫描到一个新的元素:
	首先更新 base 和 count:
		如果该元素和 base相等,那么 count 自增 1
		否则将 base 更新为当前数字,count 复位为 1
	然后更新 maxCount:
		如果 count =maxCount,那么说明当前的这个数字(basebase)出现的次数等于当前众数出现的次数,将 base 加入 answer 数组
		如果 count > maxCount,那么说明当前的这个数字( basebase)出现的次数大于当前众数出现的次数,因此,我们需要将  maxCountt 更新为  count,清空 answer 数组后将 basebase 加入  answer 数组
class Solution {
    int base, count, maxCount;
    List<Integer> answer = new ArrayList<Integer>();

    public int[] findMode(TreeNode root) {
        TreeNode cur = root, pre = null;
        while (cur != null) {
            if (cur.left == null) {
                update(cur.val);
                cur = cur.right;
                continue;
            }
            pre = cur.left;
            while (pre.right != null && pre.right != cur) {
                pre = pre.right;
            }
            if (pre.right == null) {
                pre.right = cur;
                cur = cur.left;
            } else {
                pre.right = null;
                update(cur.val);
                cur = cur.right;
            }
        }
        int[] mode = new int[answer.size()];
        for (int i = 0; i < answer.size(); ++i) {
            mode[i] = answer.get(i);
        }
        return mode;
    }

    public void update(int x) {
        if (x == base) {
            ++count;
        } else {
            count = 1;
            base = x;
        }
        if (count == maxCount) {
            answer.add(base);
        }
        if (count > maxCount) {
            maxCount = count;
            answer.clear();
            answer.add(base);
        }
    }
}
当然Mirros还可以进行前序,后序遍历
具体请参考下列文章:
	https://www.cnblogs.com/Spground/p/8536131.html
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值