树的BFS、DFS遍历

总结

今天主要是树的遍历，深度遍历（递归），广度遍历（队列）

题目

解法

		//网上版本
		if (root == null) {
            return 0;
        }
        // bfs
        Queue<TreeNode> queue = new LinkedList<>();
        int depth = 0;
        queue.add(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            depth++;
            for (int i = 0; i < size; i++) {
                TreeNode temp = queue.poll();//删除第一个元素，先进先出
                if (temp.left != null) {
                    queue.add(temp.left);
                }
                if (temp.right != null) {
                    queue.add(temp.right);
                }
            }
        }
        return depth;

//自己实现的版本
		if(root==NULL){
            return 0;
        }
        else{
            queue<TreeNode*> q;
            int depth=0;
            q.push(root);
            while(!q.empty()){
                int size=q.size();
                depth++;
                for(int i=0;i<size;i++){
                    TreeNode* temp=q.front();
                    q.pop(); 
                    if(temp->left!=NULL){
                        q.push(temp->left);
                    }
                    if(temp->right!=NULL){
                        q.push(temp->right);
                    }
                }
            }
            return depth;
        }

感悟

一般用队列来实现广度优先遍历（横着遍历root节点的每一层，然后递归下去），每遍历一层，深度+1

题目

解法

		vector<vector<int>> result;
        queue<TreeNode*> myQueue;
        if(root==NULL){
            return result;
        }
        myQueue.push(root);
        while(!myQueue.empty()){
            int size=myQueue.size();
            vector<int> temp;
            for(int i=0;i<size;i++){
                TreeNode* t=myQueue.front();
                temp.push_back(t->val);
                myQueue.pop();
                if(t->left!=NULL){
                    myQueue.push(t->left);
                }
                if(t->right!=NULL){
                    myQueue.push(t->right);
                }
            }
            result.push_back(temp);
        }
        vector<vector<int>> re;
        for(int i=result.size()-1;i>=0;i--){
            re.push_back(result[i]);
        }
        return re;

感悟

广度优先遍历的应用。需要注意的是，cpp中队列使用queue，queue有push方法，而vecotr（stack的一种）则没有，只有push-back方法；此外，queue的front可以取出队首元素，而pop只是删除首元素，不返回该元素。