JAVA用树结构实现目录系统

1使用 第一个儿子/下一兄弟表示法 来表示树

树节点定义如下：

private class TreeNode
 {
        String data;
        TreeNode firstChild;
        TreeNode nextSibling;
public TreeNode(String data, TreeNode  firstChild,
                TreeNode nextSibling) {
super();
this.data = data;
this.firstChild = firstChild;
this.nextSibling = nextSibling;
        }
    }

如下图所示：

这是一棵树

这是 上图所表示的树的第一儿子/下一兄弟表示法

如何读取指定文件夹的所有文件名，可以参考这篇文章中的方法
http://blog.csdn.net/aa8568849/article/details/52670133
在此基础上加入文件名存储进树，和输出树中的内容
程序效果如下：

代码如下：

import java.io.File;

/*
 * 
 * 
 */
public class P78_tree//存储指定文件夹所有文件名的 树类
{
private TreeNode root=new TreeNode(null, null, null);//树根（相当于链表的头指针）
public TreeNode getRoot()//获取树根
    {
return root;
    }
/*
 * 函数名：getFile
 * 作用：实现将指定文件夹的所有文件存入树中
 */
public void getFile(String path,TreeNode treeNode)
    {
        File file=new File(path);
        File[] array=file.listFiles();
        TreeNode newNode=null;
for(int i=0;i<array.length;i++)
        {
            newNode=new TreeNode(array[i].getName(),null, null);
//判断当前节点有没有firstChild，没有的话为其添加一个
if (treeNode.firstChild==null) 
            {
if (array[i].isFile())
                    treeNode.firstChild=newNode;
if (array[i].isDirectory()) 
                {
                    treeNode.firstChild=newNode;
                    getFile(array[i].getPath(), newNode);
                }
            }
//当前节点已经存在firstChild，所以后面的都是firstChild节点的兄弟
else
            {
                TreeNode p=treeNode.firstChild;
while(p.nextSibling!=null)
                    p=p.nextSibling;
if (array[i].isFile())
                    p.nextSibling=newNode;
if (array[i].isDirectory()) 
                {
                    p.nextSibling=newNode;
                    getFile(array[i].getPath(), newNode);
                }
            }   
        }
    }
/*
 * 函数名：printTree
 * 作用：输出树中的内容
 */
public void printTree(TreeNode root,int deep)
    {
if (root.data!=null) //这个if是为了区分root节点，因为root节点data=null
        {
for (int i = 0; i < deep; i++)//输出前置空格
                System.out.print(" ");
            System.out.println(root.data);
        }
//画一下图，就能理解这个两个if
if (root.firstChild!=null) 
            printTree(root.firstChild, deep+1);
if (root.nextSibling!=null)
            printTree(root.nextSibling, deep);

    }
//树节点两个指针，一个指向第一个儿子 一个指向兄弟
private class TreeNode//树节点类
 {
        String data;
        TreeNode firstChild;//第一儿子链
        TreeNode nextSibling;//下一兄弟链
//构造器
public TreeNode(String data, TreeNode  firstChild,TreeNode nextSibling)
        {
super();
this.data = data;
this.firstChild = firstChild;
this.nextSibling = nextSibling;
        }
    }
public static void main(String[] args) 
    {
        P78_tree test=new P78_tree();
        test.getFile("/Users/XXY/Desktop/test",test.getRoot());
        test.printTree(test.root, 0);
    }
}