二叉树-存储硬盘目录及文件，显示文件大小，tree命令模仿（含全部代码）

女爷i 2023-02-14 05:48 7阅读 0赞

**目录**

前言

树转二叉树

全部代码

结果截图

--------------------

# 前言 #

在cmd命令中有一个tree命令来显示目录及文件，类似下图：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70][] tree命令显示

# 树转二叉树 #

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 1][] 目录及文件结构（左）及二叉树保存（右）

其中，大写字母的表示目录，小写字母的表示文件。根据左孩子右兄弟的规律，目录下有文件或目录则有左孩子，文件没有左孩子，目录XXX下的某个目录或文件，若接下来目录XXX还有其他目录或文件，则有右孩子，否则没有。

思路：之前写过文章[二叉树基本操作函数（先、中、后、层次遍历，含全部代码）][Link 1]， 我们将其改为保存字符串的二叉树，选择先序遍历函数，写一个赋值函数，"\#"代表为NULL。另外，目录下文件或目录太多可能运行不出来，我们添加深度限制，减少递归深度，以上目录的读取顺序为：AAA BBB eee \# FFF \# \# ccc \# DDD ggg \# \# \# \#。

# 全部代码 #

#include <iostream>
    #include <io.h>
    #include <fstream>
    #include <string>
    #include <sstream>
    #include <vector>
    #define depth_limit 1   //开启深度限制
    #define max_depth 1     //最大深度2层
    using namespace std;
    //链式二叉树数据结构
    typedef struct BiTNode
    {
    	char data[100];//数据域，存放目录及文件名
    	struct BiTNode *lchild, *rchild;//左右孩子
    }BiTNode, *BiTree;
    int q = 0;
    //创建二叉树 规定数据域为"#"，则为空 先序创建
    void InitTree(BiTree &T, vector<string> files)
    {
    	if ("#" == files[q])
    	{
    		q++;
    		T = NULL;
    	}
    	else {
    		T = new BiTNode;
    		strcpy_s(T->data,strlen(files[q].c_str())+1,files[q].c_str());
    		q++;
    		InitTree(T->lchild,files);
    		InitTree(T->rchild,files);
    	}
    }
    
    // 二叉树赋值
    void TreeAssign(BiTree &T,const char *path)
    {
    	if ("#" == path) T = NULL;
    	else
    	{
    		T = new BiTNode;
    		strcpy_s(T->data,strlen(path)+1,path);
    	}
    }
    
    //二叉树结点计算
    int Count(BiTree T)
    {
    	if (T == NULL)
    		return 0;
    	else
    		return 1 + Count(T->lchild) + Count(T->rchild);
    }
    
    //先序遍历-递归
    void PreOrder(BiTree T,int depth)
    {
    	if (T != NULL)
    	{
    		for (int j = 0; j < depth; j++) 
    		{
    			 if (j == depth - 1)
    			{
    				cout << "|______";
    			}
    			else
    			cout << "      ";
    		}
    		cout << T->data<<endl;
    		PreOrder(T->lchild,depth+1);//递归先序遍历左右子树
    		PreOrder(T->rchild,depth);
    	}
    }
    
    // 读取文件大小
    size_t GetFileSize(const std::string& file_name) {
    	std::ifstream in(file_name.c_str());
    	in.seekg(0, std::ios::end);
    	size_t size = in.tellg();
    	in.close();
    	return size; //单位是：byte
    }
    
    //读取所有目录及文件 
    void getAllFiles(string path, vector<string>& files, unsigned long long &filesize, int depth)
    {
    	//文件句柄  
    	long   hFile = 0;
    	//文件信息  
    	struct _finddata_t fileinfo;  //很少用的文件信息读取结构
    	string p;  //string类很有意思的一个赋值函数:assign()，有很多重载版本
    			   //有文件或目录
    	if ((hFile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1)
    	{
    		do
    		{
    			if ((fileinfo.attrib &  _A_SUBDIR))  //比较文件类型是否是目录
    			{
    				if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
    				{
    					files.push_back(p.assign(fileinfo.name));
    					if (depth_limit&&depth == max_depth)
    					{
    						files.push_back("#");
    					}
    					else
    					{
    						getAllFiles(p.assign(path).append("/").append(fileinfo.name), files, filesize, depth + 1);
    					}
    				}
    			}
    			else
    			{
    				files.push_back(p.assign(fileinfo.name));
    				filesize += GetFileSize(p.assign(path).append("/").append(fileinfo.name));
    				//文件没有左孩子
    				files.push_back("#");
    			}
    		} while (_findnext(hFile, &fileinfo) == 0);  //寻找下一个，成功返回0，否则-1
    		_findclose(hFile);
    		//目录遍历结束
    		files.push_back("#");
    	}
    	//没有文件或目录
    	else
    	{
    		files.push_back("#");
    	}
    }
    
    //主函数
    int main() {
    	char * inPath = "F:/";
    	vector<string> files;
    	BiTree T;
    	files.push_back(inPath);
    	unsigned long long filesize = 0;
    	getAllFiles(inPath, files, filesize,0);
    	files.push_back("#");
    	TreeAssign(T, inPath);
    	TreeAssign(T->rchild, "#");
    	InitTree(T, files);
    	cout << "共" << Count(T) << "个目录及文件," << "文件总大小为" << filesize << "字节，目录及文件如下：" << endl;
    	PreOrder(T,0);
    	return 0;
    }

# 结果截图 #

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 2][] E盘2层深度显示

更多数据结构与算法实现：[数据结构（严蔚敏版）与算法的实现（含全部代码）][Link 2]

有问题请下方评论，转载请注明出处，并附有原文链接，谢谢！如有侵权，请及时联系。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200601100745454.PNG?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20200601100836437.PNG?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ==,size_16,color_FFFFFF,t_70
[Link 1]: https://blog.csdn.net/lady_killer9/article/details/86377635
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20200601101528786.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ==,size_16,color_FFFFFF,t_70
[Link 2]: https://blog.csdn.net/lady_killer9/article/details/82695895