Leetcode_#501_二叉搜索树中的众数

原题:#501_二叉搜索树中的众数

  • 二叉搜索树是一个非递减序列,所以只需要比较当前节点与前一节点pre的值即可
  • 如果pre不存在,则当前节点为第一个节点,直接将该值放入数组,计数为1
  • 如果pre存在
    • 如果pre的值等于当前的值,则计数加1
    • 如果pre的值不等于当前值,计数置为1
  • 计数与已有最值比较
    • 如果当前计数大于已有最值,则更新最值,清空数组,添加当前节点的值
    • 如果当前计数等于已有最值,则直接添加当前值到数组
int preVal = 0, curTimes = 0, maxTimes = 0;
    ArrayList<Integer> list = new ArrayList<Integer>();
    public int[] findMode(TreeNode root) {
	traversal(root);
	//list转换为int[]
	int size = list.size();
	int[] ans = new int[size];
	for(int i = 0; i < size; i++){
	    ans[i] = list.get(i);
	}
	return ans;
    }
    //二叉搜索树中序遍历是递增顺序
    public void traversal(TreeNode root){
        if(root != null){
            traversal(root.left);
            //判断当前值与上一个值的关系, 更新 curTimes 和 preVal
            if(preVal == root.val){
            curTimes++;
            }else{
            preVal = root.val;
            curTimes = 1;
            }
            //判断当前数量与最大数量的关系, 更新 list 和 maxTimes
            if(curTimes == maxTimes){
            list.add(root.val);
            }else if(curTimes > maxTimes){
            list.clear();
            list.add(root.val);
            maxTimes = curTimes;
            }
            traversal(root.right);
        }
    }
原创文章 50 获赞 1 访问量 2911

猜你喜欢

转载自blog.csdn.net/u014642412/article/details/105952577
今日推荐