PAT A 1115. Counting Nodes in a BST (30)

只要是树的题都不会很难，简单的建立排序树，再层次遍历。

测试点5通不过是因为未考虑只有一个结点的情况。

#include <cstdio>
#include <vector>
#include <cstdlib>
#include <queue>
using namespace std;

struct node{
	int key;
	node *lchild,*rchild;
};

int keys[1002]={0};
int levelnode[1002]={0};
queue<node*> Q;

void create_bst(node *&T,const int &i,const int &n){
	if(i<n){
		if(T==NULL){
			T=(node *)malloc(sizeof(node));
			T->key=keys[i];
			T->lchild=T->rchild=NULL;
		}
		else if(keys[i]<=T->key)create_bst(T->lchild,i,n);
		else if(keys[i]>T->key)create_bst(T->rchild,i,n);
	}
}

int bfs(node *T){
	int level=0;
	Q.push(T);
	node *rear=T;
	while(!Q.empty()){
		node *tmp=Q.front();
		Q.pop();
		++levelnode[level];
		if(tmp->lchild!=NULL)Q.push(tmp->lchild);
		if(tmp->rchild!=NULL)Q.push(tmp->rchild);
		if(rear==tmp){//最好使用指向节点的指针来判断是否进入下一层，而不要使用关键字相等来判断，因为题中很明显会出现相关的关键字，
			++level;//从而可以避免莫名其妙的BUG，当然这个题不会有这种情况。
			rear=Q.back();
		}
	}
	return level-1;
}

int main(void){
	//freopen("in.log","r",stdin);
	int n;
	scanf("%d",&n);
	for(int i=0;i<n;++i)scanf("%d",&keys[i]);
	node *T=NULL;
	for(int i=0;i<n;++i)create_bst(T,i,n);
	int level=bfs(T);
	printf("%d + %d = %d\n",levelnode[level],levelnode[level-1],levelnode[level]+levelnode[level-1]);
	return 0;
}