蓝桥杯-----ALGO-27 FBI树

 

问题描述

　　我们可以把由“0”和“1”组成的字符串分为三类：全“0”串称为B串，全“1”串称为I串，既含“0”又含“1”的串则称为F串。
　　FBI树是一种二叉树，它的结点类型也包括F结点，B结点和I结点三种。由一个长度为2N的“01”串S可以构造出一棵FBI树T，递归的构造方法如下：
　　1)T的根结点为R，其类型与串S的类型相同；
　　2)若串S的长度大于1，将串S从中间分开，分为等长的左右子串S1和S2；由左子串S1构造R的左子树T1，由右子串S2构造R的右子树T2。
　　现在给定一个长度为2N的“01”串，请用上述构造方法构造出一棵FBI树，并输出它的后序遍历序列。

输入格式

　　第一行是一个整数N（0 <= N <= 10），第二行是一个长度为2N的“01”串。

输出格式

　　包括一行，这一行只包含一个字符串，即FBI树的后序遍历序列。

样例输入

3
10001011

样例输出

IBFBBBFIBFIIIFF

数据规模和约定

　　对于40%的数据，N <= 2；
　　对于全部的数据，N <= 10。
　　注：
　　[1] 二叉树：二叉树是结点的有限集合，这个集合或为空集，或由一个根结点和两棵不相交的二叉树组成。这两棵不相交的二叉树分别称为这个根结点的左子树和右子树。
　　[2] 后序遍历：后序遍历是深度优先遍历二叉树的一种方法，它的递归定义是：先后序遍历左子树，再后序遍历右子树，最后访问根。

• 首先这是一道关于数据结构中二叉树的问题

把输入的“01”串转变为一棵树，这些输入的“01”串将作为所生成树的树叶

所以我们可以从树叶开始往上生成这个二叉树

如题，先建立底层树叶的节点，然后只需再后面生成根节点，把子节点连接起来

所以我们需要一个记录树节点的nodeList数组和一个记录“01”串值的int型数组

然后通过判断数组的值，就可以为对应nodeList上的树节点赋值

然后每次循环判断是否到生成根节点的那一步

如果是就退出，然后就可以通过数的根节点后续遍历输出答案

树的类声明

class Node{
	private String data;
	private Node left;
	private Node right;
	
	Node(String d){
		this.data = d;
	}

	public String getData() {
		return data;
	}

	public void setData(String data) {
		this.data = data;
	}

	public Node getLeft() {
		return left;
	}

	public void setLeft(Node left) {
		this.left = left;
	}

	public Node getRight() {
		return right;
	}

	public void setRight(Node right) {
		this.right = right;
	}		
}

 下面是实现由底向上建树的核心代码

while(totalNode >= flag) {
	//由底往上建树
	for(int i = 0; i < totalNode; i+=flag) {
		if(nodeList[i] == null) {
			nodeList[i] = new Node("");
		}
		else {
			Tmp = new Node("");
			Tmp.setLeft(nodeList[i]);
			Tmp.setRight(nodeList[i+flag/2]);
			nodeList[i] = Tmp;					//更新树,往上搭起来
					
			if(Tmp.getLeft().getData() != Tmp.getRight().getData() || Tmp.getLeft().getData() == "F") {
				arrNum[i] = -1;
			}
			else {
				if(Tmp.getLeft().getData() == "I") {
					arrNum[i] = 1;
				}
				else {
					arrNum[i] = 0;
				}
			}
		}
				
		//往里面写数据
		switch (arrNum[i]) {
		case 1:
			nodeList[i].setData("I");
			break;
		case 0:
			nodeList[i].setData("B");
			break;
		default:
			nodeList[i].setData("F");
			break;
		}
	}
	flag *= 2;			//更改flag的移动位置
}

完整代码 

package c3_2019_3_6_FBITree;

import java.util.Scanner;

class Node{
	private String data;
	private Node left;
	private Node right;
	
	Node(String d){
		this.data = d;
	}

	public String getData() {
		return data;
	}

	public void setData(String data) {
		this.data = data;
	}

	public Node getLeft() {
		return left;
	}

	public void setLeft(Node left) {
		this.left = left;
	}

	public Node getRight() {
		return right;
	}

	public void setRight(Node right) {
		this.right = right;
	}		
}

public class Main {
	
	private static Node tree;
	private static int N;
	private static int arr[];
	
	public static void main(String[] args) {
				
		Scanner sc = new Scanner(System.in);
		
		N = sc.nextInt();
		
		arr = new int[(int)Math.pow(2, N)];
		
		String tmpStr = sc.next();			//获取输入的10串
		
		for(int i = 0; i < arr.length; i++) {
			arr[i] = (int)(tmpStr.charAt(i) - '0');
		}
		
		tree = CreateTree(arr);
		
//		System.out.println("中序遍历:");
//		printTreeMid(tree);
//		System.out.println();
//		System.out.println("先序遍历:");
//		printTreeFir(tree);
//		System.out.println();
//		System.out.println("后序遍历:");
		printTreeLast(tree);
		
	}
	
	private static Node CreateTree(int arrNum[]) {
		int flag = 1;									//方便后面差速移动数组
		int totalNode = arrNum.length; 					//当前所有节点，当节点只剩下一个时树就生成了
		Node nodeList[] = new Node[arrNum.length];		//树节点的数组
		Node Tmp;										//零时存储新建的节点			
		while(totalNode >= flag) {
			//由底往上建树
			for(int i = 0; i < totalNode; i+=flag) {
				if(nodeList[i] == null) {
					nodeList[i] = new Node("");
				}
				else {
					Tmp = new Node("");
					Tmp.setLeft(nodeList[i]);
					Tmp.setRight(nodeList[i+flag/2]);
					nodeList[i] = Tmp;					//更新树,往上搭起来
					
					if(Tmp.getLeft().getData() != Tmp.getRight().getData() || Tmp.getLeft().getData() == "F") {
						arrNum[i] = -1;
					}
					else {
						if(Tmp.getLeft().getData() == "I") {
							arrNum[i] = 1;
						}
						else {
							arrNum[i] = 0;
						}
					}
				}
				
				//往里面写数据
				switch (arrNum[i]) {
				case 1:
					nodeList[i].setData("I");
					break;
				case 0:
					nodeList[i].setData("B");
					break;
				default:
					nodeList[i].setData("F");
					break;
				}
			}
			flag *= 2;			//更改flag的移动位置
		}
		
		return nodeList[0];
		
	}
	
	private static void printTreeMid(Node tree) {
		if(tree.getLeft() != null) {
			printTreeMid(tree.getLeft());
		}
		System.out.print(tree.getData());
		if(tree.getRight() != null) {
			printTreeMid(tree.getRight());
		}
	}
	
	private static void printTreeFir(Node tree) {
		System.out.print(tree.getData());
		if(tree.getLeft() != null) {
			printTreeFir(tree.getLeft());
		}
		if(tree.getRight() != null) {
			printTreeFir(tree.getRight());
		}
	}
	
	private static void printTreeLast(Node tree) {
		if(tree.getLeft() != null) {
			printTreeLast(tree.getLeft());
		}
		if(tree.getRight() != null) {
			printTreeLast(tree.getRight());
		}
		System.out.print(tree.getData());
	}
}