1115 Counting Nodes in a BST

• 题目
• 题意
• 代码解析
• AC代码
• 参考

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805355987451904

题意

将给定数字放入二叉搜索树中，并输出最低两层的结点数量及其总和

代码解析

建树的insert函数就是常规流程

判断结点数有两种方式——dfs和bfs，这两个方法都能AC

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef struct node* tree;
struct node{
	int data;
	tree left,right;
};
tree insert(tree a,int t)
{
	if(!a)
	{
		a=new node();
		a->data=t;
		a->left=a->right=NULL;
		return a;
	}
	else if(t<=a->data)
		a->left=insert(a->left,t);
	else
		a->right=insert(a->right,t);
	return a;
}
vector<int> ans(1005,0);
int pos=0;
void bfs(tree a)
{
	queue<tree> q;
	q.push(a);
	tree last=q.back();
	while(q.size())
	{
		tree b=q.front();
		q.pop();
		ans[pos]++;
		if(b->left) q.push(b->left);
		if(b->right) q.push(b->right);
		if(b==last) 
		{
			pos++;
			last=q.back();
		}
	}
}
//void dfs(tree a,int depth)
//{
//	if(a==NULL)
//	{
//		pos=max(pos,depth);
//		return;
//	}
//	ans[depth]++;
//	dfs(a->left,depth+1);
//	dfs(a->right,depth+1);
//}
int main()
{
	int n,t;
	tree a=NULL;
	cin>>n;
	while(n--)
	{
		cin>>t;
		a=insert(a,t);
	}
	bfs(a);
//	dfs(a,0);
	int x=ans[pos-1],y=ans[pos-2];
	printf("%d + %d = %d",x,y,x+y);
}

参考

dfs部分参考了1115. Counting Nodes in a BST (30)-PAT甲级真题（二叉树的遍历，dfs）