[LeetCode 111] - 二叉树的最小深度 (Minimum Depth of Binary Tree)

问题

给出一棵二叉树，找出它的最小深度。

最小深度是指从根节点沿着最短路径下降到最近的叶子节点所经过的节点数。

 

初始思路

不难看出又是一个需要层次遍历二叉树的题目，只要在112基础上作出简单修改即可得出答案。

class Solution
    {
    public:
        int minDepth(TreeNode *root)
        {
            if(!root)
            {
                return 0;
            }
            
            treeLevel_[0].clear();
            treeLevel_[1].clear();
            
            depth_ = 0;
            
            bool flag = false;
            treeLevel_[0].push_back(root);
            
            while(!treeLevel_[flag].empty())
            {
                ++depth_;
                for(auto iter = treeLevel_[flag].begin(); iter != treeLevel_[flag].end(); ++iter)
                {
                    if(!(*iter)->left && !(*iter)->right)
                    {
                        return depth_;
                    }
                    
                    if((*iter)->left)
                    {
                        treeLevel_[!flag].push_back((*iter)->left);
                    }
                    
                    if((*iter)->right)
                    {
                        treeLevel_[!flag].push_back((*iter)->right);
                    }
                }
                
                treeLevel_[flag].clear();
                
                flag = !flag;
            }
            
            return depth_;

        }
        
    private:
        std::vector<TreeNode*> treeLevel_[2];
        int depth_;
    };