[팁] 두 갈래 나무의 여러 가지 흔적

차례차례 두루 다니다
중서 역행
교체 버전1:
class Solution {
public:
    vector inorderTraversal(TreeNode* root) {
        stack S;
        vector result;
        
        TreeNode *cur = root;
        
        while(true){
            if(cur != NULL){
                S.push(cur);
                cur = cur->left;
            }else if(!S.empty()){
                cur = S.top();  S.pop();
                result.push_back(cur->val);
                cur = cur->right;
            }else{
                break;
            }
        }
        return result;
    }
};

다음 순서를 반복합니다.
교체 버전1:
stack S;     // 
        
TreeNode *cur = root;
S.push(cur);
        
while(!S.empty()){
    if(S.top()->left != cur && S.top()->right != cur){  // 
        gotoHLVFL(S);
    }
    // 
    cur = S.top();  S.pop();
    //visit(cur)
}


void gotoHLVFL(stack &S){
    TreeNode *cur;
    while((cur = S.top()) != NULL){
        if(cur->left != NULL){
            if(cur->right != NULL) S.push(cur->right);
            S.push(cur->left);
            // cur = cur->left;
        }else{
            S.push(cur->right);
            // cur = cur->right;
        }
    }
    S.pop();
}

교체 버전 2:
vector postOrder(TreeNode *root)
{
    vector res;
    if(root == NULL) return res;

    TreeNode *p = root;
    stack sta;
    sta.push(p);
    sta.push(p);
    while(!sta.empty())
    {
        p = sta.top(); sta.pop();
        if(!sta.empty() && p==sta.top())
        {
            if(p->right) sta.push(p->right), sta.push(p->right);
            if(p->left) sta.push(p->left), sta.push(p->left);
        }
        else
            res.push_back(p->val);
    }
    
    return res;
}

모든 노드에 대해 두 번 눌러서 순환체에서 매번 하나의 노드가 p에게 주어진다. 만약에 p가 창고의 머리 결점과 같다면 p의 아이들은 아직 조작된 적이 없기 때문에 그 아이들을 창고에 넣어야 한다. 그렇지 않으면 p에 접근해야 한다.즉, 첫 번째 팝업, p의 아이를 창고에 눌러 넣고, 두 번째 팝업, p에 접근한다. 

좋은 웹페이지 즐겨찾기