leetcode102

也是一道经典题目，继续利用上题的队列进行广度遍历，然后标记每一层的最后一个节点，一个在入队的时候标记，可以提前知道下一层的节点个数，一个在出队的时候标记，可以根据入队时的标记知道是否是这一层最后一个节点，然后进行分层输出。

/**
 * 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<vector<int>> levelOrder(TreeNode* root) {
    vector<vector<int> >res;
    vector<int>mid;
    if(root==NULL)return res;
    queue<TreeNode*>Node;
    Node.push(root);
    int num=0,count=0,flag=1;
    while(!Node.empty())
    {
        TreeNode* p=Node.front();
        mid.push_back(p->val);
        Node.pop();
        num++;
        if(p->left){
            Node.push(p->left);
            count++;
        }
        if(p->right){
            Node.push(p->right);   
            count++;
        }
        if(flag==num)
        {
            res.push_back(mid);
            mid.clear();
            num=0;
            flag=count;
            count=0;
        }
    }
    return res;
    }
};