algorithm - 广度优先搜索算法变体

Question

所以我是这个编程的新手，我对学习它的这一部分有疑问，对于两个变化问题，几乎代码是相似的，有一点改变，但我不太清楚。

Node 类类型是标准的基于树的结构以及下一个自引用

连接级别顺序兄弟：

class Solution {
public:
    Node* connect(Node* root) {
        
        return helper(root);
        
    }
    
    Node* helper(Node* root)
    {
        if(root==NULL)           
            return root;
        
        queue<Node*>queue;
        queue.push(root);
        
        while(!(queue.empty()))
        {
            int levelsize=queue.size();
            Node* prev=NULL;
            
            for(int i = 0  ; i < levelsize ; i++)
            {
                Node* current=queue.front();
                queue.pop();
                
                if(prev!=NULL)
                    prev->next=current;
                
                prev=current;
                
                if(current->left!=NULL)
                    queue.push(current->left);
                if(current->right!=NULL)
                    queue.push(current->right);
                
            }   
        }

        return root;
        
    }
    
};

连接所有级别订单兄弟：

class Solution {
public:
    Node* connect(Node* root) 
    {
       return helper(root);
    }
    
    Node* helper(Node* root)
    {
        if(root==NULL)           
            return root;
        
        queue<Node*>queue;
        queue.push(root);
        
        while(!(queue.empty()))
        {
            int levelsize=queue.size();
            Node* prev=NULL;
           
                Node* current=queue.front();
                queue.pop();
                
                if(prev!=NULL)
                    prev->next=current;
                
                prev=current;
                
                if(current->left!=NULL)
                    queue.push(current->left);
                if(current->right!=NULL)
                    queue.push(current->right);
                
          
          
        }

        return root;
        
    }
    
};

标签： algorithmbreadth-first-search