[LeetCode] 102.Binary Tree Level Order Traversal

102题

一、问题描述

二、问题分析

问题描述是很清晰明了的，就是要求我们广度优先遍历二叉树。广度优先顾名思义，是指对于一张图(无论是有向图还是无向图)，从一个节点出发，首先访问所有它可以到达的顶点，之后再访问对于第一次访问的所有节点可到达的下一些节点，以此类推。对于BFS，最合适的方法是用一个队列queue来储存它访问的节点。每次访问一个节点就把它pop出来，并将其可以访问的未访问节点添加到队列尾部，并置为已访问过。伪代码如下图：

三、问题求解

针对问题分析，以下是c++源代码。这里，我用了一个map来记录每个节点与出发点之间的最短路径(层数)是多少；而如何判断什么模式访问到了新的一层呢？那就是要看每一个节点的层数与result里面向量个数(已访问的层数)的大小关系了。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> result;
        if (root == NULL) {
        	return result;
        }
        queue<TreeNode*> Q;
        Q.push(root);
        map<TreeNode*, int> layer;
        layer[root] = 1;

        while (!Q.empty()) {
        	TreeNode* temp = Q.front();
        	Q.pop();
        	int layerNum = layer[temp];
            if (layerNum > result.size()) {
        		vector<int> another;
                another.push_back(temp->val);
        		result.push_back(another);
        	}
            else {
                result.back().push_back(temp->val);
            }
        	if (temp->left != NULL) {
        		Q.push(temp->left);
        		layer[temp->left] = layerNum + 1;
        	}
        	if (temp->right != NULL) {
        		Q.push(temp->right);
        		layer[temp->right] = layerNum + 1;
        	}
        }
        return result;
    }
};

四、问题改进

有没有更简单的方法？或者说可以不用图来存储每个节点的层数呢？看了一下别人的优质代码，结果是肯定的。即外层还是用while来判断队列是否有元素，而内层多加一个for循环，循环的次数就是每层的节点个数，这样做完一次循环就相当访问了一层。然而随着新节点的加入以及旧节点的删除，queue的size是不固定的，如何确定每层节点的个数？其实只要在每个循环一开始看看queue的size即可，因为这时所有节点都位于将访问的下一层。代码如下：

class Solution {
public:
    vector<vector<int> > levelOrder(TreeNode *root) {
    	vector<vector<int>> result;
    	if (!root) return result;
    	queue<TreeNode* > Q;
    	Q.push(root);
    	while (!Q.empty()) {
    		vector<int> level;
    		int size = Q.size(); //注意，这里的size不能在循环里面写，因为size是会变的
    		for (int i = 0; i < size; i++) {
    			TreeNode* node = Q.front();
    			Q.pop();
    			level.push_back(node->val);
    			if (node->left) Q.push(node->left);
    			if (node->right) Q.push(node->right);
    		}
    		result.push_back(level);
    	}
    	return result;
    }
};