LeetCode 103. Binary Tree Zigzag Level Order Traversal

LeetCode 102. Binary Tree Level Order Traversal

需要加上bool need_reverse判断当前行(level)的数字是否需要reverse。

代码：

class Solution 
{
public:
    vector<vector<int> > zigzagLevelOrder(TreeNode *root) 
    {
    	vector<vector<int>> ret;
        vector<TreeNode*> same_level_node[2];
		same_level_node[0].push_back(root);
		bool need_reverse = false;

        while (same_level_node[0].empty() == false)
        {
        	vector<int> num;
        	same_level_node[1].clear();
        	
        	for (size_t i = 0; i < same_level_node[0].size(); ++ i)
        	{
        		if (same_level_node[0][i] != NULL)
        		{
        			num.push_back( same_level_node[0][i]->val );
        			same_level_node[1].push_back( same_level_node[0][i]->left );
        			same_level_node[1].push_back( same_level_node[0][i]->right );        			
        		}
        	}
        	if (need_reverse == true)
        	{
        		reverse(num.begin(), num.end());
        	}
        	need_reverse = !need_reverse;
            if (num.size() != 0)
            {
            	ret.push_back(num);
            }
        	same_level_node[0] = same_level_node[1];
        }

        return ret;
    }
};