首页 > 其他分享 >【LeetCode二叉树#15】二叉搜索树中的众数(递归中序遍历)

【LeetCode二叉树#15】二叉搜索树中的众数(递归中序遍历)

时间:2023-03-02 13:22:05浏览次数:52  
标签:count 15 cur 中序 二叉树 当前 maxCount 节点 指针

二叉搜索树中的众数

力扣题目链接(opens new window)

给定一个有相同值的二叉搜索树(BST),找出 BST 中的所有众数(出现频率最高的元素)。

假定 BST 有如下定义:

  • 结点左子树中所含结点的值小于等于当前结点的值
  • 结点右子树中所含结点的值大于等于当前结点的值
  • 左子树和右子树都是二叉搜索树

例如:

给定 BST [1,null,2,2],

501. 二叉搜索树中的众数

返回[2].

提示:如果众数超过1个,不需考虑输出顺序

进阶:你可以不使用额外的空间吗?(假设由递归产生的隐式调用栈的开销不被计算在内)

思路

初见思路是:直接把二叉树遍历一遍,得到结果数组,在用map进行频率统计,返回出现最多的那个即可

但是,像上面这样处理的话,统计频率那块的代码不太好写(需要自定义比较函数)

所以还是用 中序遍历+递归 的方式去做

不同的是,这里也要引入 二叉搜索树的最小绝对差 中的双指针的技巧

即在遍历过程中设置两个指针,一个指向前一节点,一个指向当前节点

在本题中,通过前一节点与当前节点的比较,来判断一个元素是否出现了多次,并统计其出现次数

这里还需要设置两个计数变量,一个count用来记录当前遍历值的出现次数,一个maxCount用于记录出现过的最大count。

当需要更新maxCount时,同时记录下当前的节点值并清空上次记录的节点值。

代码分析

还是先写递归函数

因为我们要遍历 整颗二叉树 ,且不需要立刻返回判断值,所以递归函数的输入参数是根节点且没有返回值。根据递归的中序遍历不难写出以下代码:

//定义计数变量
int maxCount = 0;
int count = 0;
//定义前一节点指针
TreeNode* pre = NULL;

void searchBST(TreeNode* cur){
    //确定终止条件
    if(cur == NULL) return;
    //左
    searchBST(cur->left);
    //中,比较前一节点和当前节点并统计当前节点值的出现次数
    
    //右
    searchBST(cur->right);
    
    return;//void也要写return
}

在 中 的处理逻辑里,我们引入双指针

一共有下面3中情况需要count去记录

  • 前一节点指针为空时,当前节点指针指向第一个节点,因此count需要记1
  • 前一节点指针不为空时,且当前节点指针值与前一节点值不相等,count需要记1(当前遍历值只出现了一次)
  • 当前节点指针值与前一节点值相等,count需要累加记录

当记录完成,更新当前节点与前一节点的指针即可

//定义计数变量
int maxCount = 0;
int count = 0;
//定义前一节点指针
TreeNode* pre = NULL;

void searchBST(TreeNode* cur){
    //左
    searchBST(cur->left);
    //中,比较前一节点和当前节点并统计当前节点值的出现次数
    if(pre == NULL){//当前节点指针指向第一个节点
        count = 1;
    }else if(pre->val == cur->val){//当前节点指针值与前一节点值相等
        count++;
    }else{//前一节点指针不为空时,且当前节点指针值与前一节点值不相等
        count = 1;
    }
    pre = cur;//更新当前节点与前一节点的指针
    
    //右
    searchBST(cur->right);
    
    return;//void也要写return
}

得到count之后,我们需要去更新maxCount,也有两种情况:

  • count与maxCount相同,说明有多个元素,它们都出现了多次并且次数相同(即多个众数的情况)
  • count大于maxCount,说明有某个元素的出现次数更多,该元素将成为新的众数

情况一种我们只需将当前遍历的节点的值再加入结果数组即可

情况二则需要先清空之前结果数组保存的值,再加入新的众数(即当前遍历的节点的值)

上述过程也在 中 的处理逻辑里完成

//定义计数变量
int maxCount = 0;
int count = 0;
//定义前一节点指针
TreeNode* pre = NULL;
//定义结果数组
vector<int> res;

void searchBST(TreeNode* cur){
    //左
    searchBST(cur->left);
    //中,比较前一节点和当前节点并统计当前节点值的出现次数
    if(pre == NULL){//当前节点指针指向第一个节点
        count = 1;
    }else if(pre->val == cur->val){//当前节点指针值与前一节点值相等
        count++;
    }else{//前一节点指针不为空时,且当前节点指针值与前一节点值不相等
        count = 1;
    }
    //比较并更新maxCount,记录结果值
    if(count == maxCount){//出现多个众数
        res.push_back(cur->val);
    }else if(count > maxCount){//出现新的众数
        //更新最大出现次数
        maxCount = count;
        //清空结果数组
        res.clear();
        //更新众数
        res.push_back(cur->val);
    }
    pre = cur;
    
    //右
    searchBST(cur->right);
    return;//void也要写return
}

完整代码

class Solution {
private:
    //定义计数变量
    int maxCount = 0;
    int count = 0;
    //定义前一节点指针
    TreeNode* pre = NULL;
    //定义结果数组
    vector<int> res;

    void searchBST(TreeNode* cur){
        //确定终止条件
        if(cur == NULL) return;
        //左
        searchBST(cur->left);
        //中,比较前一节点和当前节点并统计当前节点值的出现次数
        if(pre == NULL){//当前节点指针指向第一个节点
            count = 1;
        }else if(pre->val == cur->val){//当前节点指针值与前一节点值相等
            count++;
        }else{//前一节点指针不为空时,且当前节点指针值与前一节点值不相等
            count = 1;
        }
        pre = cur;

        //比较并更新maxCount,记录结果值
        if(count == maxCount){//出现多个众数
            res.push_back(cur->val);
        }else if(count > maxCount){//出现新的众数
            //更新最大出现次数
            maxCount = count;
            //清空结果数组
            res.clear();
            //更新众数
            res.push_back(cur->val);
        }
        //右
        searchBST(cur->right);
        return;//void也要写return
    }
public:
    vector<int> findMode(TreeNode* root) {
        //res.clear();//清理一下结果数组

        searchBST(root);//调用递归
        return res;//返回结果数组
    }
};

标签:count,15,cur,中序,二叉树,当前,maxCount,节点,指针
From: https://www.cnblogs.com/DAYceng/p/17171470.html

相关文章

  • AtCoder Regular Contest 155
    期末考完复健,补一下一个月前打的ARC当时赛后9秒过D,太痛了,第一次体验这种只能说,幸好当时要打的时候感觉状态不行,就unrated了比赛的状况是:A不知道哪错了;C不会;D博弈DP原......
  • 不写代码,拿百万年薪!ChatGPT提示工程或造就15亿码农大军
    新智元报道编辑:编辑部近日,据称是史上首位的提示工程师接受了《华盛顿邮报》采访,为我们揭秘了这个年薪百万的全新职业。ChatGPT爆火之后,带火了一项「......
  • TZOJ 4767: 二叉树遍历
    描述  给定一颗二叉树,要求输出对二叉树进行先序和后序遍历所得到的序列。本题假设二叉树的结点数不超过1000。  输入  输入数据分为多组,第一行是测试数......
  • AcWing 1562. 微博转发
    微博被称为中文版的Twitter。微博上的用户既可能有很多关注者,也可能关注很多其他用户。因此,形成了一种基于这些关注关系的社交网络。当用户在微博上发布帖子时,他/她的......
  • P8631 [蓝桥杯 2015 国 AC] 切开字符串 题解
    P8631[蓝桥杯2015国AC]切开字符串题解前言看到这题没有人写题解,就打算写一篇。这也是蒟蒻的第一篇题解。前置知识\(manacher\),\(SA\)。如果不会可以转至mana......
  • CF15D-Map题解
    题目传送门题意:有一个\(n\timesm\)的矩阵,每个格子有一个权值。每次操作会选择一个\(x\timesy\)的矩形区域,花费为“每个位置的权值减去最小权值”之和,区域之间不能......
  • EDU-CFR-115-Div-2解题报告
    赛时AC3道,补题做出来一道A.ComputerGame\(Problem\)有一个\(2\timesn\)的01矩阵,1为障碍,你要从\((1,1)\)走到\((2,n)\),每一步只能向右、上、下、右上、右下走,问......
  • C. Maximum Set[数学] [*1300-*1500]
    C.MaximumSet[数学][*1300-*1500]题目链接点我题意:一个集合是漂亮的,如果他的每一个元素都是集合中其他元素的倍数或者因子给定你一个\(l\)和\(r\)让你找出在\(......
  • ubuntu20.04二进制部署PostgreSQL 15.2
    创建文件存储库配置echo"debhttp://apt.postgresql.org/pub/repos/apt$(lsb_release-cs)-pgdgmain">/etc/apt/sources.list.d/pgdg.list导入存储库签名密钥wget......
  • iOS15.4以上NSDateformatter12小时制日期格式问题及解决
    最近有客户反应App点击提交按钮报错的问题,经过分析发现问题是因为时间格式的问题导致的,客户系统iOS15.4以上,系统时间格式设置的是12小时制如下代码NSDate*date=[NSDat......