PAT 1115

#include<cstdio>
#include<algorithm>
#include<vector>
#include<iostream>
#include<queue>
using namespace std;
const int maxn = 1010;
int in[maxn];
struct Node
{
	Node* lchild;
	Node* rchild;
	int data;
	int depth;
};
void insert(Node* &root,int a)
{
	if (root == NULL)
	{
		root = new Node;
		root->lchild = root->rchild = NULL;
		root->data = a;
		return;
	}
	if (a <= root->data)
		insert(root->lchild, a);
	else
		insert(root->rchild, a);

}
int dep[maxn] = {0};
void bfs(Node* root)
{
	if (root == NULL) return;
	queue<Node*> q;
	q.push(root);
	root->depth = 0;
	
	while (!q.empty())
	{
		Node* top = q.front();
		q.pop();
		dep[top->depth]++;
		if (top->lchild!=NULL)
		{
			q.push(top->lchild);
			top->lchild->depth = top->depth+1;
		}
		if (top->rchild != NULL)
		{
			q.push(top->rchild);
			top->rchild->depth = top->depth+1;
		}
		
	}
}
int main()
{
	int n;
	cin >> n;
	Node* root = NULL;
	for (int i=0;i<n;i++)
	{
		cin >> in[i];	
		insert(root,in[i]);
	}
	bfs(root);
	int i;
	for (i=0;i<n;i++)
	{
		if (dep[i] == 0)
			break;
	}
	i--;
	if (i >= 1)
	{
		cout << dep[i] << " + " << dep[i - 1] << " = " << dep[i] + dep[i - 1] << endl;
	}
	else
		cout << dep[i] << " + " << '0' << " = " << dep[i] << endl;
	system("pause");
	return 0;

}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值