408 14年41题wpl

/*
41.(13 分)二叉树的带权路径长度(WPL)是二叉树中所有叶结点的带权路径长度
之和。给定一棵二叉树T,采用二叉链表存储，结点结构如下:
left
weight
right
其中叶结点的weight域保存该结点的非负权值。设root为指向T的根结点的指针，请设计
求T的WPL的算法，要求:
1)给出算法的基本设计思想。
2)使用C或C++语言，给出二叉树结点的数据类型定义。
3)根据设计思想，采用C或C++语言描述算法，关键之处给出注释。
*/

#include <stdio.h>
#include <stdlib.h>

#define BiType int
#define ElementType BiTree
typedef struct BiNode{
    BiNode* left;
    BiType data;
    BiNode* right;
}BiNode,*BiTree;

typedef struct LinkNode{
    ElementType data;
    LinkNode* next;
}LinkNode,*LinkList;

typedef struct Queue{
    LinkNode* front;
    LinkNode* rear;
};



void initQueue(Queue &Q){
    LinkList L=(LinkNode*) malloc(sizeof(LinkNode));
    Q.front=Q.rear=L;
}

void EnQueue(Queue &Q,LinkNode* ele){
    LinkNode *NewNode=(LinkNode*) malloc(sizeof(LinkNode));
    NewNode=ele;
    NewNode->next=NULL;
    Q.rear->next=NewNode;
    Q.rear=Q.rear->next;
}

void DeQueue(Queue &Q,LinkNode* &delEle){
    if(Q.rear==Q.front){
        printf("the queue is empty");
    }
    else{
        delEle=Q.front->next;
        free(Q.front);
        Q.front=delEle;
    }
}

bool isEmpty(Queue Q){
    if(Q.front==Q.rear){
        return true;
    }
    else{
        return false;
    }

}

void BuildTree(BiTree &root,Queue &Q){
    root=(BiNode*) malloc(sizeof(BiNode));
    root=Q.front->next->data;
    BiNode *biNode=root;
    //Q.front指向元素为空
    LinkNode *ChildNode=Q.front->next->next;
    LinkNode* delEle;
    while (ChildNode){
        if (!biNode->left){
            biNode->left=ChildNode->data;
            ChildNode=ChildNode->next;
        }
        //需加入判断，防止结束的时候右节点没有孩子，将NULL插入到右节点发生错误
        if (!biNode->right&&ChildNode){
            biNode->right=ChildNode->data;
            ChildNode=ChildNode->next;
        }
        if (biNode->right&&biNode->right){
            DeQueue(Q,delEle);
            biNode=Q.front->next->data;
        }
    }
}

void PreOrder(BiTree tree,int &wpl,int level){
    if (tree){
        level++;
        printf("%dlevel=%d\n",tree->data,level);
        if(!tree->left&&!tree->right){
            wpl=wpl+level*tree->data;
        }
        PreOrder(tree->left,wpl,level);
        PreOrder(tree->right,wpl,level);
    }
}

int main() {
    Queue Q;
    BiTree tree;
    initQueue(Q);
    for(int i=0;i<10;i++){
        LinkNode *Lnode=(LinkNode*) malloc(sizeof(LinkNode));
        BiNode*  Binode=(BiNode*) malloc(sizeof(BiNode));
        Binode->data=i;
        Binode->left=Binode->right=NULL;
        Lnode->data=Binode;
        EnQueue(Q,Lnode);
    }
    BuildTree(tree,Q);

    int WPL=0;
    int level=0;
    PreOrder(tree,WPL,level);
    printf("WPL=%d",WPL);
    return 0;
}