CTreeCtrl类的递归使用

CTreeCtrl类的递归使用


作者：

本文是对第20期中"遍历文件夹并建成目录树"一文的补充。

CTreeCtrl是可是化编程中很实用的一个类，可以用于目录结构、层次结构、属性结构,尤其是在显示文件目录结构时更是应用广泛。看了第20期北京林业大学的李少杰朋友的一篇"遍历文件夹并建成目录树"，觉得深有感触，初学VC时确实CTreeCtrl类很难掌握；至于对"树的遍历"，也是数据结构的一个难点。李朋友的那篇文章解决了如果用CTreeCtrl显示目录结构，在他程序中，没有能够完善解决如何递归未知层数的目录的问题，(在CtreeCtrl的使用和函数的递归调用上有不足之处)，我想对这个问题作一点补充。

我的这个程序可以搜任意层目录，其输出结果如下：




下面我谈谈怎么编写这个程序：
首先，也是用MFC AppWizard[exe]生成一个dialog base的框架，在上面放一个Tree控件，并创建一个成员函数m_Tree来指向这个控件。
之后，编写递归搜索函数

      void AddFile( CString StrPath, HTREEITEM faItem );

最后，在OnInitDialog()中调用AddFile，这样在程序已开始运行就显示Tree。

详细的内容请参考附带的源代码。

下面，我将着重讲讲如何编写AddFile函数。对于文件目录结构的遍历，选用递归算法是不错的。用递归方法对树进行遍历常用的有深度优先和广度优先两种搜索方法，由于我们要遍力整个树，所以选用深度优先的算法是不错的。因为在这里，深度优先和广度优先的时间复杂度一样，但是深度优先比较节省堆栈资源。以下，就是采用深度优先搜索的AddFile函数，我将在程序中作进一步说明。

// 递归搜索文件路径，采用深度优先搜索法

void CFileTreeDlg::AddFile(CString StrPath, HTREEITEM faItem )

				//StrPath为传递过来的目录层次，本次函数调用中搜索的文件都是它的下一层的。

				//faItem为传递过来的Tree节点，本次函数调用中添加的Tree节点都是它的子节点。

{

	CFileFind OneFile;

	CString FName, DirName;

	BOOL BeWorking; 

	HTREEITEM NewItem;	



	DirName = StrPath "\\*.*";

	BeWorking = OneFile.FindFile( DirName );

	while ( BeWorking ) {  		//BeWorking非零，指找了文件或目录



		//查找同级的目录

		BeWorking = OneFile.FindNextFile();



		if ( OneFile.IsDirectory() && !OneFile.IsDots() )        //如果查找的结果是目录又不是".."或"."

		{	

			//向Tree1中添加目录;

			DirName = OneFile.GetFilePath();

			FName = OneFile.GetFileTitle(); 

			//IDC_TREE1

			NewItem = m_Tree.InsertItem( FName, faItem );    //NewItem取得节点，其目的是为了下一层中

									 //添加节点方便，递归时把它传过去。

			//进入下一层递归调用。

			AddFile(DirName, NewItem);

		}



		//退出递归时，到了这里！！！

		if ( !OneFile.IsDirectory() && !OneFile.IsDots() )       //如果查找结果是文件

		{

			//向Tree1中添加文件

			FName = OneFile.GetFileTitle();			 //注意这里用的是GetFileTitle,因为

									 //这里是添加文件。

			m_Tree.InsertItem( FName, faItem );

		}



		

	}// end of while

	

	OneFile.Close();						//记着用完CFileFild实例要关闭

}           

以上程序在 VC6 professional sp5 上通过,操作系统是 Win2000 professional。