PTA二叉搜索树的2层结点统计

L2-3 二叉搜索树的2层结点统计

二叉搜索树或者是一棵空树，或者是具有下列性质的二叉树：若它的左子树不空，则左子树上所有结点的值均小于或等于它的根结点的值；若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值；它的左、右子树也分别为二叉搜索树。

将一系列数字按给定顺序插入一棵初始为空的二叉搜索树，你的任务是统计结果树中最下面 2 层的结点数。

输入格式：输入在第一行给出一个正整数 N (≤1000)，为插入数字的个数。第二行给出 N 个 [−1000,1000] 区间内的整数。数字间以空格分隔。

输出格式：在一行中输出最下面 2 层的结点总数。

输入样例：

9 25 30 42 16 20 20 35 -5 28

输出样例：

6

本题思路，根据定义建树，按照顺序插入树，一层一层的判断，如果大于就进入左子树，小于进入右子树，建树完以后就简单了，通过层次遍历，得到最后两层节点数即可

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.io.StreamTokenizer;
import java.util.LinkedList;
import java.util.Queue;

public class Main二叉搜索树的2层结点统计 {
	public static void main(String[] args) throws IOException {
		StreamTokenizer x = new StreamTokenizer(new BufferedReader(new InputStreamReader(System.in)));
		PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
		x.nextToken();
		int n=(int)x.nval;
		Dat root=null;
		for(int i=0;i<n;i++) {
			x.nextToken();
			root=gz(root,(int)x.nval);
		}
		out.println(cs(root));
		out.flush();
	}
	public static Dat gz(Dat p,int i){
		if(p==null) {
			p=new Dat(i);
			return p;
		}
		if(i>p.data)
			p.right=gz(p.right,i);
		else
			p.left=gz(p.left,i);
		return p;
	}
	public static int cs(Dat p) {
		Queue<Dat> pp=new LinkedList<Dat>();
		pp.add(p);
		int l1=0,l2=0;
		while(!pp.isEmpty()) {
			l2=l1;
			l1=0;
			int n=pp.size();
			while(n--!=0) {
				l1++;
				Dat q=pp.poll();
				if(q.left!=null)
				pp.add(q.left);
				if(q.right!=null)
				pp.add(q.right);
			}
		}
		return l1+l2;
	}
}
class Dat{
	int data;
	Dat left,right;
	public Dat(int d) {
		this.data=d;
		this.left=null;
		this.right=null;
	}
}