假定某二叉树以链接形式(每个结点包括三个字段,left,data,right)存储,请设计一个算法,求该二叉树的宽度(一棵二叉树的宽度系指:在该二叉树的各层上,具有结点树最多的那一层上的结点总数)。
解题思路:ggt在层次遍历过程中用一个变量记录当前层结点数目与现存最大值比较决定是否更新。
核心改动:在层次遍历的基础上将一层的结点的孩子都入队即可得到一层的所有结点。
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
string qx="1245309";//完全二叉树测试数据
string zx="4251039";
//string qx="12485309";//非完全二叉树测试数据
//string zx="48251039";
const int MaxSize=100;
typedef struct Tree//二叉树结构定义
{
int data;
Tree *left;
Tree *right;
} *BiTree;
BiTree build(string qx,string zx)//建树
{
if(qx.size()==0)return NULL;
Tree *root=(Tree *)malloc(sizeof(Tree));
root->data=qx[0]-'0';
int pos=zx.find(qx[0]);
root->left=build(qx.substr(1,pos),zx.substr(0,pos));
root->right=build(qx.substr(pos+1),zx.substr(pos+1));
return root;
}
typedef struct LinkNode//队列结点
{
BiTree data;
LinkNode *next;
}LinkNode,*Link;
typedef struct Queue//链队列
{
LinkNode *front;
LinkNode *rear;
}*queue;
void InitQueue(Queue &Q)
{
Q.front=Q.rear=(LinkNode*)malloc(sizeof(LinkNode));
Q.front->next=NULL;
}
bool isEmpty(Queue Q)
{
if(Q.front==Q.rear) return true;
else return false;
}
void EnQueue(Queue &Q,BiTree x)
{
LinkNode *p=(LinkNode*)malloc(sizeof(LinkNode));
p->data=x;
p->next=Q.rear->next;
Q.rear->next=p;
Q.rear=p;
}
bool DeQueue(Queue &Q,BiTree &x)
{
if(Q.front==Q.rear) return false;
LinkNode *p=Q.front->next;
x=p->data;
Q.front->next=p->next;
if(Q.rear==p) Q.rear=Q.front;
return true;
}
int QueueLength(Queue Q)//求队长
{
int len=0;
LinkNode *q=Q.front->next;
while(q)
{
len++;
q=q->next;
}
return len;
}
int width(BiTree T)//层次遍历求二叉树的宽度
{
int maxwidth=0; //记录最大宽度
BiTree x;
Queue Q;
InitQueue(Q);
BiTree p=T;
EnQueue(Q,p);
while(!isEmpty(Q))
{
int wid=QueueLength(Q);
if(wid>maxwidth)
{
maxwidth=wid;//更新最大宽度
}
printf("width===%d\n",wid);//在中间测试一下
while(wid--)//在层次遍历的基础上将一层的结点的孩子都入队即可得到一层的所有结点
{
DeQueue(Q,x);
if(x->left!=NULL)
EnQueue(Q,x->left);
if(x->right!=NULL)
EnQueue(Q,x->right);
}
}
return maxwidth;
}
int main()
{
BiTree root=build(qx,zx);//建树
int maxw=width(root);
printf("宽度为%d",maxw);
}