1115 Counting Nodes in a BST (30 分)

最新推荐文章于 2024-10-10 22:59:02 发布

Bloo8m

最新推荐文章于 2024-10-10 22:59:02 发布

阅读量128

点赞数

分类专栏：笔记

本文链接：https://blog.csdn.net/qq_51791303/article/details/123055375

版权

深度优先链表算法

笔记专栏收录该内容

24 篇文章 0 订阅

订阅专栏

n=1000，用链表建树。
注意建树的过程，不太会。
建树完树之后，dfs统计每个深度的节点个数。

在这里插入图片描述

#include<bits/stdc++.h>
using namespace std;
int n,ma=0;
vector<int> v(1005);
struct node{
	int val;
	struct node *l,*r; 
};
node* create(node *root,int val){
	if(root==NULL){
		root=new node();
		root->val=val;
		root->l=root->r=NULL;
	}
	else if(root->val>=val)
		root->l=create(root->l,val);
	else
		root->r=create(root->r,val);
	return root;
}
void dfs(node *root,int de){
		if(root==NULL) return ;
		v[de]++;
		ma=max(ma,de);
		dfs(root->l,de+1);
		dfs(root->r,de+1);
}
int main(){
	cin>>n;
	node *root=NULL;
	for(int i=0;i<n;i++){
		int x;
		cin>>x;
		root=create(root,x);
	}
	dfs(root,0);
	printf("%d + %d = %d\n",v[ma],v[ma-1],v[ma-1]+v[ma]);
	return 0;
}