LeetCode 501. 二叉搜索树中的众数（中序遍历）

例如：
给定 BST [1,null,2,2],

   1
    \
     2
    /
   2
返回[2].

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

class Solution {
public:
    vector<int> findMode(TreeNode* root) {
        stack<TreeNode*> stk;
        TreeNode *tp;
        vector<int> ans;
        int prev, maxlen = 0, curlen = 0;
        bool flag = true;
        while(root || !stk.empty())
        {
        	while(root)
        	{
        		stk.push(root);
        		root = root->left;
        	}
        	tp = stk.top();
        	stk.pop();
        	root = tp->right;
        	if(flag)//只进来一次
        	{
        		prev = tp->val;
        		flag = false;
        		curlen = 1;
        		maxlen = 1;
        		ans.push_back(tp->val);
        		continue;
        	}
        	if(tp->val == prev)
        	{
        		curlen++;
        		if(curlen > maxlen)
        		{	//碰见频率大的
        			maxlen = curlen;
        			ans.clear();//清空频率小的
        			ans.push_back(tp->val);
        		}
        		else if(curlen == maxlen)
        			ans.push_back(tp->val);//频率相同的都进去
        	}
        	else//与前一个不相等
        	{
        		prev = tp->val;
        		curlen = 1;
        		if(curlen == maxlen)//curlen只可能==maxlen为1的情况
        			ans.push_back(tp->val);
        	}
        }
        return ans;
    }
};