Python怎么找到二叉搜索樹中的所有mode

本篇內容介紹了“Python怎么找到二叉搜索樹中的所有mode”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！

題目：找到二叉搜索樹中的所有mode(出現最頻繁的元素)。

思路：這里定義的二叉搜索樹的一個結點的左子樹中所有結點的值都小于或等于該結點的值，右子樹則相反，大于或等于。于此同時，follow up說讓我們不用除了遞歸中的隱含棧之外的額外空間，因此不能使用哈希表。由于是二叉搜索樹，那么我們中序遍歷出來的結果就是有序的，我們只需要比較前后兩個元素是否相等，就能統計某個元素出現的次數，因為相同的元素可定是都在一起的。我們需要一個結點變量pre來記錄上一個遍歷到的結點，然后mx記錄最大的次數，cnt計數當前元素出現的次數。中序遍歷的時候，如果pre不為空，說明當前不是第一個結點，我們和之前一個結點比較，如果相等，則cnt自增1，如果不等，cnt重置1。如果此時cnt大于mx，那么我們清空結果res，并把當前結點值放入結果res，如果cnt等于mx，那么我們直接將當前結點值加入結果res，然后mx賦值為cnt。最后我們把pre更新為當前結點。如果pre為空，說明當前結點是根結點。那么我們新建一個結點并賦上當前結點值。

Language : cpp

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */class Solution {public:vector<int> findMode(TreeNode* root) {if (!root) return {};vector<int> res;
        TreeNode *now = root, *pre = NULL;stack<TreeNode*> s;int mx = 0, cnt = 1;while (!s.empty() || now) {while (now) {             //中序遍歷，左中右。將每個結點的左子樹入棧s.push(now);
                now = now->left;
            }
            now = s.top(); s.pop(); //取棧頂元素if (pre) {              //判斷當前元素和上一個元素值是否一樣，一樣cnt計數加一cnt = (now->val == pre->val) ? cnt + 1 : 1;
            }                       //如果cnt大于等于mx，說明當前元素重復次數大于之前最大的重復元素的次數，需要將新結果入結果棧。if (cnt >= mx) {if (cnt > mx) res.clear();
                res.push_back(now->val);
                mx = cnt;
            }if (!pre) pre = new TreeNode(now->val);
            pre->val = now->val;
            now = now->right;
        }return res;
    }
};

Language : python

# Definition for a binary tree node.# class TreeNode(object):#     def __init__(self, x):#         self.val = x#         self.left = None#         self.right = Noneclass Solution(object):def findMode(self, root):"""
        :type root: TreeNode
        :rtype: List[int]
        """res = []
        s = []
        now = root
        pre = Nonemx, cnt = 0, 1while s or now:while now:
                s.append(now)
                now = now.left
            now = s.pop(len(s) - 1)if pre:if now.val == pre.val:
                    cnt = cnt + 1else:
                    cnt = 1if cnt >= mx:if cnt > mx:del res[:]
                res.append(now.val)
                mx = cntif not pre:
                pre = TreeNode(now.val)
            pre.val = now.val
            now = now.rightreturn res

“Python怎么找到二叉搜索樹中的所有mode”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！