每日一題演算法：2020年9月24日 [二叉搜尋樹中的眾數] findMode

2020年9月24日 二叉搜尋樹中的眾數 findMode

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int[] findMode(TreeNode root) {

    }
}

解題思路：

題目中都提示我們使用遞迴來解決這個問題了，那肯定是選擇遞迴了。

首先我們使用會用到額外空間的方式，計數法去實現，寫一個額外的函式去遞迴二叉樹，將所有數字出現的次數進行統計，然後輸出出現次數最多的那些眾數。

程式碼實現：使用最愚蠢也最簡單的方式去實現。

HashMap<Integer,Integer> map=new HashMap<>();

public int[] findMode(TreeNode root) {

    findmore(root);

    int maxNum=0;

    List<Integer> list = new ArrayList<>();

    for (Integer integer : map.keySet()) {

        if (map.get(integer) > maxNum){
            maxNum = map.get(integer);
            list = new ArrayList<>();
            list.add(integer);
        }else if (map.get(integer) == maxNum){
            list.add(integer);
        }

    }

    int[] res=new int[list.size()];
    for (int i=0;i<list.size();i++) {
        res[i]=list.get(i);
    }

    return res;
}


public void findmore(TreeNode root){

    if (root==null)
        return;

    if (map.get(root.val)==null){

        map.put(root.val,1 );

    }
    else
        map.put(root.val,map.get(root.val)+1 );

    findmore(root.right);
    findmore(root.left);

}