Average of Levels in Binary Tree

Given a non-empty binary tree, return the average value of the nodes on each level in the form of an array.
Example 1:

Input:
    3
   / \
  9  20
    /  \
   15   7
Output: [3, 14.5, 11]
**Explanation:**
The average value of nodes on level 0 is 3,  on level 1 is 14.5, and on level 2 is 11. Hence return [3, 14.5, 11].
**Note:**
The range of node's value is in the range of 32-bit signed integer.

解析:感觉这个利用nullptr的解答方法还是技巧性很强的,程序员的思路真的很神奇~可能是我还没入门吧。。。贴上c++的解答感受一下:

#include
#include
#include 
using namespace std;

template<class T>                  //模版结构体
struct TreeNode
{
    T data;
    TreeNode  *Lchild, *Rchild; //节点的左子树和右子树
    //可选择参数的默认构造函数
};


template <class T> //模版方法,声明一个模板,虚拟类型名为T(数据类型),尖括号中是模板的参数列表,关键字class后面是类型参数,后面没有分号

void createBinaryTree(TreeNode *&root)  //传递指针的引用
{
    TreeNode* p = root;//指向T类对象的指针
    T nodeValue;
    cin >> nodeValue;
    if (nodeValue == -1)
    {
        root = NULL;
    }
    else
    {
        root = new TreeNode();            //构造一个节点
        root->data = nodeValue;
        createBinaryTree(root->Lchild);    //递归构造左子树(*p).Lchild,即root所指向的对象中的成员或函数 
        createBinaryTree(root->Rchild);    //递归构造右子树
    }
}

template<class T>
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> ret;
        double sum = 0, count = 0;
        queue*> q;
        q.push(root);
        q.push(nullptr);
        while (!q.empty()) {
            TreeNode* t = q.front();
            q.pop();
            if (t == nullptr) {
                ret.push_back(sum / count);
                sum = count = 0.0;
                if (!q.empty()) q.push(nullptr);
            }
            else {
                sum += t->data;
                ++count;
                if (t->Lchild) q.push(t->Lchild);
                if (t->Rchild) q.push(t->Rchild);
            }
        }
        return ret;
    }


int main()
{
    TreeNode<int> * rootNode = NULL;
    cout << "请输入每个结点,回车确认,并以-1结束:\n";
    createBinaryTree(rootNode);//3 9 -1 -1 20 15 -1 -1 7 -1 -1
    averageOfLevels(rootNode);
    return 0;
}

其中,对模板参数和声明虚拟类型的使用还是不太熟练,对vector和queue的使用也要加强学习。附上一个c++各容器的比较:
http://www.cnblogs.com/For-her/p/3918761.html

你可能感兴趣的:(c++)