zoukankan      html  css  js  c++  java
  • Binary Tree Zigzag Level Order Traversal

    题目的意思是z字形遍历二叉树,代码有点冗余希望大牛么赐教,大概意思是要不从左到右遍历存入栈中,要不从右至左存入栈中,为了统一操作新建了一个类似头节点的树节点。

    /**
     * Definition for binary tree
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
     * };
     */
    class Solution {
    public:
        vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
            // Start typing your C/C++ solution below
            // DO NOT write int main() function
            bool flag = true;
            vector<vector<int> > result;
            if(root==NULL)
            return result;
            vector<TreeNode*> stack;
            vector<TreeNode*> temp;
            vector<int> layer;
            TreeNode *head = (TreeNode*)malloc(sizeof(TreeNode));
            head->left = root;
            head->right = NULL;
            stack.push_back(head);
            while(!stack.empty())
            {
                temp.clear();
                layer.clear();
                if(flag)
                {
                   for(vector<TreeNode*>::iterator iter = stack.end()-1;iter>=stack.begin();iter--)
                   {
                       if((*iter)->left!=NULL)
                       {
                           temp.push_back((*iter)->left);
                           layer.push_back((*iter)->left->val);
                       }
                       if((*iter)->right!=NULL)
                       {
                           temp.push_back((*iter)->right);
                           layer.push_back((*iter)->right->val);
                       }
                   }
                }
                else
                {
                    for(vector<TreeNode*>::iterator iter = stack.end()-1;iter>=stack.begin();iter--)
                   {
                       if((*iter)->right!=NULL)
                       {
                           temp.push_back((*iter)->right);
                           layer.push_back((*iter)->right->val);
                       }
                       if((*iter)->left!=NULL)
                       {
                           temp.push_back((*iter)->left);
                           layer.push_back((*iter)->left->val);
                       }
                   }
                }
                stack = temp;
                if(!layer.empty())
                result.push_back(layer);
                flag = !flag;
            }
            free(head);
            head =NULL;
            return result;
        }
    };
  • 相关阅读:
    day4-1
    day3-1
    day1-3
    day2-1
    day1-2
    day1 1
    对象的高度整合
    类和数据类型
    对象的绑定方法
    python总结
  • 原文地址:https://www.cnblogs.com/727713-chuan/p/3318233.html
Copyright © 2011-2022 走看看