
递归:
1/** 2 * Definition for a binary tree node. 3 * struct TreeNode { 4 * int val; 5 * TreeNode *left; 6 * TreeNode *right; 7 * TreeNode(int x) : val(x), left(NULL), right(NULL) {} 8 * }; 9 */ 10class Solution { 11 12 13 14public: 15 vector<int> findMode(TreeNode* root) { 16 17 18 19 unordered_map<int, int> count; //纪录每一个节点数字的个数(键值对) 20 vector<int> result; //结果 21 int maxCount=0; // 最大个数,也就是众数 22 search(root,count,maxCount); 23 for (auto v: count) { 24 25 26 // 因为众数不止一个 27 if(v.second==maxCount){ 28 29 30 31 result.push_back(v.first); 32 } 33 } 34 return result; 35 } 36 //遍历整棵树 37 void search(TreeNode* root,unordered_map<int, int> &count,int &maxCount){ 38 39 40 41 if(root==nullptr) return;// 叶子节点结束 42 count[root->val]++; // 纪录加一 43 if(count[root->val]>maxCount) maxCount=count[root->val]; //纪录最大的个数(众数) 44 search(root->left,count,maxCount);// 遍历左节点 45 search(root->right,count,maxCount);//遍历右节点 46 } 47};

本文同步分享在 博客“战 胜”(CSDN)。
如有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一起分享。