1115. Counting Nodes in a BST (30)

题目详情：https://www.patest.cn/contests/pat-a-practise/1115

感觉很简单，但是没有写temp->left=NULL 和 temp->right=NULL 导致程序写错了，想了好长时间！特此纪念！

#include <iostream>#include <malloc.h>#include <vector>using namespace std;typedef struct BSTree{    struct BSTree* left;    int data,level;//level存储该节点所在的层次    struct BSTree* right;}BSTree;BSTree* insert(BSTree* root,int value){    if(root==NULL)//找到插入位置    {        BSTree* temp=(BSTree*)malloc(sizeof(BSTree));        temp->left=NULL;temp->right=NULL; //没写这两句话居然错了        temp->data=value;        return temp;    }    else if( root->data >= value )//根节点的值较大，则需要到左子树上去找插入位置    {        root->left=insert(root->left,value);    }    else//root->data < value,去右子树上找插入位置    {        root->right=insert(root->right,value);    }    return root;}vector<int> levelOrder(BSTree* root){    vector<int> num_level;//存储每层节点的个数    vector<BSTree*> queue;//存储树节点的队列    BSTree* temp;    root->level=1;//根节点的层次设置为1    int front=0;//队列中未访问的第一个节点    queue.push_back(root);    while( front!=queue.size() )    {        temp=queue[front];        if(temp->left)        {            temp->left->level=temp->level+1; //层次加1            queue.push_back(temp->left);//把左节点加入到队列中        }        if(temp->right)        {            temp->right->level=temp->level+1;//层次加1            queue.push_back(temp->right);//把右节点加入到队列中        }        if(num_level.size()>=temp->level)//temp节点所在的节点层次已经在num_level中            num_level[temp->level -1]++;//那么直接将层次的节点个数加1        else//否则把1添加到最后，代表该层次中只有一个节点            num_level.push_back(1);        front++;    }    return num_level;}int main(){    int n,temp;    BSTree* root=NULL;    scanf("%d",&n);    for( int i=0;i<n;++i )    {        scanf("%d",&temp);        root=insert(root,temp);    }    vector<int> num_level=levelOrder(root);//可能只有一层节点,测试用例没有这种可能    int length=num_level.size();    cout<<num_level[length-1]<<" + "<<num_level[length-2]<<" = "<<num_level[length-1]+num_level[length-2]<<endl;}