小工具:找出下一層的子節點

queue<TreeNode*> find_level(queue<TreeNode*>q){
    	queue<TreeNode*>res;
    	while(!q.empty()) {
    		TreeNode* top = q.front();
    		q.pop();

    		if(top==NULL) continue;
    	
    		res.push(top->left);
    		res.push(top->right);
    	}
    	return res;
    }

這是將NULL結點也放進去的版本(上面)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章