算法学习 - 求二叉树的宽度

二叉树的宽度

二叉树的宽度定义为

• 整个二叉树各层节点数，其中最大的值为这个二叉树的宽度。

所以二叉树的第一层就是1（根节点）。

代码实现(C++)

代码实现比较简单，树的遍历一般用递归比较方便。

////  main.cpp//  TreeWidth////  Created by Alps on 15/3/11.//  Copyright (c) 2015年 chen. All rights reserved.//// achieve the binary tree width.#include <iostream>using namespace std;#ifndef MAXDEEP#define MAXDEEP 10 //定义树的最大深度#endifstruct Node{ //定义树节点    int val;    struct Node * right;    struct Node * left;    Node(int v,Node* r,Node* l): val(v), right(r), left(l) {}};typedef Node* Tree;int deepth = 0; //遍历的层int width[MAXDEEP] = {0}; //存放各层宽度的数组void TreeWidth(Tree t){    if (t == NULL) {        return;    }    if (deepth == 0) {        width[0] = 1;    }    if (t->left != NULL) {        width[deepth+1] += 1;        deepth += 1;        TreeWidth(t->left);    }    if (t->right != NULL) {        width[deepth+1] += 1;        deepth+=1;        TreeWidth(t->right);    }    deepth-=1;}int main(int argc, const char * argv[]) {    Node n1(1,NULL,NULL);    Node n2(2,NULL,NULL);    Node n3(3,NULL,NULL);    Node n4(4,&n1,&n2);    Node n5(5,&n3,NULL);    Node n6(6,&n4,&n5);    Tree t = &n6;    TreeWidth(t);    for (int i = 0; i < MAXDEEP; i++) {        printf("%d ",width[i]);    }    // insert code here...    std::cout << "Hello, World!\n";    return 0;}