2016.6.24——vector<vector<int>>【Binary Tree Level Order Traversal】

Binary Tree Level Order Traversal

本题收获：

1.vector<vector<int>>的用法

　　vector<vector<int> >注意<int>后面的空格，vector<vector<int>>表示的是二位向量。

　　输出格式（后面代码），不知道大小时，在vector中用push_back(vector<int>())

2.树用迭代

　　题目：

　　Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

　　For example:
　　Given binary tree [3,9,20,null,null,15,7],

　　    3
 　　  / 
  　　9  20
  　　  /  
   　　15   7

　　return its level order traversal as:

　　[
  　　[3],
  　　[9,20],
  　　[15,7]
　　]

　　思路：

　　　　：1.用vector<vector<int>>输出二位数组 2.迭代。

　　代码：

vector<vector<int>> ret;

void buildVector(TreeNode *root, int depth)
{
    if(root == NULL) return;
    if(ret.size() == depth)
 7         ret.push_back(vector<int>());　　　　//depth的设置很巧妙

    ret[depth].push_back(root->val);
    buildVector(root->left, depth + 1);
    buildVector(root->right, depth + 1);
}

vector<vector<int> > levelOrder(TreeNode *root) {
    buildVector(root, 0);
    return ret;
}

　　vector<vector<int>>输出格式：

int n = res.size();
    for (int i = 0; i < n; i++)
    {
        int m = res[i].size();                //注意vector<vector<int> >的输出，以及size是怎么设定 
        for (int j = 0; j < m; j++)
        {
            cout << res[i][j] << " ";
        }
        cout << endl;                        //输出格式 cout << endl的位置
    }

　　全部测试代码：

// Binary Tree Level Order Traversal.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include "iostream"
#include "malloc.h"
#include "vector"
using namespace std;

struct TreeNode
{
    int val;
    TreeNode *left, *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL){};
};

class MyClass
{
public:
    vector<vector<int> > res;
    vector<vector<int>> levelOrder(TreeNode* root)
    {
        
        buildvector(root, 0);
        return  res;
    }

    void buildvector(TreeNode* root, int depth)
    {
        if (root == NULL) return;
        if (res.size() == depth)
        {
            res.push_back(vector<int>());
        }

        res[depth].push_back(root->val);
        buildvector(root->left, depth + 1);
        buildvector(root->right, depth + 1);
    }
};



void creatTree(TreeNode* &T)
{
    int data;
    cin >> data;
    if (data == -1)
    {
        T = NULL;
    }
    else
    {
        T = (TreeNode*)malloc(sizeof(TreeNode));
        T->val = data;
        creatTree(T->left);
        creatTree(T->right);
    }
}

int _tmain(int argc, _TCHAR* argv[])
{
    TreeNode* root = NULL;
    creatTree(root);
    vector<vector<int> > res;
    MyClass solution;
    res = solution.levelOrder(root);
    int n = res.size();
    for (int i = 0; i < n; i++)
    {
        int m = res[i].size();                //注意vector<vector<int> >的输出，以及size是怎么设定 
        for (int j = 0; j < m; j++)
        {
            cout << res[i][j] << " ";
        }
        cout << endl;                        //输出格式
    }
    system("pause");
    return 0;
}

　　3
   / 
  9  20
    /  
   15   7
上面的树，在本题作为输入为（3 9 -1 -1 20 15 -1 -1 7 -1 -1） 
-1代表后面没有子节点。