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)
          ret.push_back(vector<int>());　　　　//depth的设置很巧妙

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

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

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

 int n = res.size();
     for (int i = ; i < n; i++)
     {
         int m = res[i].size();                //注意vector<vector<int> >的输出，以及size是怎么设定
         for (int j = ; 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, );
         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 + );
         buildvector(root->right, depth + );
     }
 };

 void creatTree(TreeNode* &T)
 {
     int data;
     cin >> data;
     if (data == -)
     {
         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 = ; i < n; i++)
     {
         int m = res[i].size();                //注意vector<vector<int> >的输出，以及size是怎么设定
         for (int j = ; j < m; j++)
         {
             cout << res[i][j] << " ";
         }
         cout << endl;                        //输出格式
     }
     system("pause");
     return ;
 }

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