求二叉树中指定节点的深度

package job.huawei;

import java.util.Scanner;

public class TreeHigh {

    /**
     * 输入字符串如a1b2c2d3e3f3g4构成一颗二叉树数字表示前一个字符所在的层，最多不超过9层
     * 求指定节点的深度
     */
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner cin=new Scanner(System.in);
        String str=cin.next();
        String findChar=cin.next();
        cin.close();
        char tree[]=new char[512];
        createTree(str,tree);
        for(int i=0;i<findChar.length();i++){
            System.out.print(getNodeHigh(tree,str,findChar.charAt(i))+" ");
        }        
    }
    /**
     * 根据输入的字符串，顺序存储树
     * @param str
     * @param tree
     */
    public static void createTree(String str,char tree[]){
        int k=0,count=0,j=0;
        for(int i=1;i<str.length();i+=2){
            j=Integer.parseInt(String.valueOf(str.charAt(i)));//取某节点所在层

            if(i>1)
            {
                count=str.charAt(i)!=str.charAt(i-2)?0:count+1;//判断是否在同一层，同一层则count+1，否则count=0

            }
            k=(int)Math.pow(2, j-1)+count;////计算节点的存储位置

            tree[k]=str.charAt(i-1);
        }
    }
    /**
     * 求指定节点在树中的深度
     * @param tree
     * @param str
     * @param x
     * @return
     */
    public static int getNodeHigh(char tree[],String str,char x) {
        int index=str.indexOf(String.valueOf(x));//判断该字符是否在树中，不在则直接返回

        if(index<0) return 0;
        
        int d=Integer.parseInt(String.valueOf(str.charAt(index+1))),level=0;//获取节点所在层次   
        
        while(2*d<tree.length&&tree[2*d]!=tree[0]){
            level++;
            d=2*d;
        }
        return level+1;
    }
}