数据结构课程设计---文件目录结构显示（c++版）

最新推荐文章于 2024-04-15 20:45:26 发布

暗夜的彩虹

最新推荐文章于 2024-04-15 20:45:26 发布

阅读量3.1k

点赞数 5

分类专栏：数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/qq_15742375/article/details/85324043

版权

数据结构专栏收录该内容

4 篇文章 0 订阅

订阅专栏

文件目录结构显示

【问题描述】

要求在给出Unix或windows下目录和文件信息的前提下，编程实现将其排列成一棵具有一定缩进的树。

【基本要求】

在给定的磁盘目录下，参照下表分层缩进输出文件目录结构

|_*/usr

|_*mark

| |_hw.c

| |_*course

|_aa.txt

|_*alex

|_hw.c

|_*usr

ps：代码中读取文件的函数和保存文件信息的结构体等在网上都能查到，这里只分析算法实现部分。

【算法设计】：

用结构体（node）构建一棵多叉树，每一个节点代表一个文件夹。利用结构体数组模仿链表，在每一个节点里设置一个数组保存子节点的位置（即子节点在结构体数组中的下标）。每一个节点node里包含整变量deep，记录当前文件深度；string类型name代表当前文件夹的名字；string类型my_path表示当前文件夹路径；vector<string>file_name动态数组存取当前文件夹下的文件名字；整型数组d[100]存取当前目录下的子文件夹节点的位置，整型变量dlen存取d数组的长度。

struct node//多叉树的每一个节点
{
    string name,my_path;
    vector<string>file_name;
    int d[100],dlen;
    int deep;
}a[1000000];

①构建多叉树 void build(int p)

用递归实现，递归函数传参为当前的节点下标。每一次递归遍历当前节点（文件夹目录）所有的子文件，子文件夹。如果遍历到文件夹，则将路径和名字保存到一个新的节点中，并在当前节点中保存新节点的下标，新节点的deep等于当前节点的deep+1，新节点创建完成进行递归，传参为新节点的下标。如果遍历到文件，将文件名保存到当前节点的vector<string>file_name中。

/*
构建多叉树----递归实现
*/
void build(int p)
{
    long oFile = 0;
    struct _finddata_t fileinfo;  //保存文件信息的结构体
    string str;
    if ((oFile=_findfirst(str.assign(a[p].my_path).append("\\*").c_str(),&fileinfo)) != -1) {
        do{
            if((fileinfo.attrib&_A_SUBDIR)){//比较文件类型是否是文件夹
                if(strcmp(fileinfo.name,".")!=0&&strcmp(fileinfo.name,"..")!=0) {
                    a[alen].name=fileinfo.name;
                    a[alen].deep=a[p].deep+1;
                    a[alen].dlen=0;
                    a[alen].my_path=str.assign(a[p].my_path).append("\\").append(fileinfo.name);
                    alen++;
                    a[p].d[a[p].dlen++]=alen-1;
                    build(alen-1);
                }
            }
            else
                a[p].file_name.push_back(fileinfo.name);
        } while(_findnext(oFile,&fileinfo)==0);
        _findclose(oFile);
    }
}

②遍历输出多叉树 void print(int p)

递归实现，递归函数参数为当前节点的下标。对于每一次递归，先输出当前节点的名字，deep为缩进数；之后遍历递归当前节点的子文件夹，当所有子文件夹都遍历完，输出当前节点下的文件名（vector<string>file_name），缩进数为deep+1。

/*
遍历输出多叉树----递归实现
*/
void print(int p)
{
    if(p>=alen)        return;
    //输出当前节点的name，缩进为   a[p].deep
    for(int i=0;i<a[p].deep;i++)
        printf("\t");//    \t为一个缩进
    printf("|_*%s\n",a[p].name.c_str());
    //遍历递归当前节点的子文件夹
    for(int i=0;i<a[p].dlen;i++)
        print(a[p].d[i]);
    //输出当前节点的文件，缩进为     a[p].deep+1
    for(int i=0;i<a[p].file_name.size();i++){
        for(int j=0;j<a[p].deep+1;j++)
            printf("\t");
        printf("|_%s\n",a[p].file_name[i].c_str());
    }
}

【程序测试】：略。

【全部代码】：

/*
    课设名称：文件目录结构显示
    学校：烟台大学
    学院：计算机与控制工程学院
    班级：计171
    姓名：武传涛
    代码语言：C++
    编译器：CodeBlocks
*/
#include <iostream>
#include<stdio.h>
#include<stdlib.h>
#include<io.h>
#include<fstream>
#include<tchar.h>
#include<string.h>
#include <shlobj.h>
#include<vector>
using namespace std;
struct node//多叉树的每一个节点
{
    string name,my_path;
    vector<string>file_name;
    int d[100],dlen;
    int deep;
}a[1000000];
int alen=1;
void array_init()
{
    for(int i=0;i<=alen;i++){
        memset(a[i].d,0,sizeof a[i].d);
        a[i].dlen=0;
        a[i].deep=0;
        a[i].file_name.clear();
        a[i].name=a[i].my_path="";
    }
    alen=1;
}
/*
构建多叉树----递归实现
*/
void build(int p)
{
    long oFile = 0;
    struct _finddata_t fileinfo;  //保存文件信息的结构体
    string str;
    if ((oFile=_findfirst(str.assign(a[p].my_path).append("\\*").c_str(),&fileinfo)) != -1) {
        do{
            if((fileinfo.attrib&_A_SUBDIR)){//比较文件类型是否是文件夹
                if(strcmp(fileinfo.name,".")!=0&&strcmp(fileinfo.name,"..")!=0) {
                    a[alen].name=fileinfo.name;
                    a[alen].deep=a[p].deep+1;
                    a[alen].dlen=0;
                    a[alen].my_path=str.assign(a[p].my_path).append("\\").append(fileinfo.name);
                    alen++;
                    a[p].d[a[p].dlen++]=alen-1;
                    build(alen-1);
                }
            }
            else
                a[p].file_name.push_back(fileinfo.name);
        } while(_findnext(oFile,&fileinfo)==0);
        _findclose(oFile);
    }
}
/*
遍历输出多叉树----递归实现
*/
void print(int p)
{
    if(p>=alen)        return;
    //输出当前节点的name，缩进为   a[p].deep
    for(int i=0;i<a[p].deep;i++)
        printf("\t");//    \t为一个缩进
    printf("|_*%s\n",a[p].name.c_str());
    //遍历递归当前节点的子文件夹
    for(int i=0;i<a[p].dlen;i++)
        print(a[p].d[i]);
    //输出当前节点的文件，缩进为     a[p].deep+1
    for(int i=0;i<a[p].file_name.size();i++){
        for(int j=0;j<a[p].deep+1;j++)
            printf("\t");
        printf("|_%s\n",a[p].file_name[i].c_str());
    }
}
int main()
{
    while(1){
        string str;
        bool flag=false;
        //清屏操作
        system("CLS");
        printf("请输入一个目录路径或者输入\"open\"来手动选择目录(输入\"end\"结束程序)\n\n");
        getline(cin,str);
        //如果输入end，结束程序！
        if(str=="end")      break;//、
        //初始化
        array_init();
        if(str=="open"){
            //显示打开文件夹窗口，并获取选择的目录路径。
            TCHAR szBuffer[MAX_PATH] = {0};
            BROWSEINFO bi;
            ZeroMemory(&bi,sizeof(BROWSEINFO));
            bi.hwndOwner = NULL;
            bi.pszDisplayName = szBuffer;
            bi.lpszTitle = _T("选择文件夹:");
            bi.ulFlags = BIF_RETURNFSANCESTORS;
            LPITEMIDLIST idl = SHBrowseForFolder(&bi);
            //如果成功选择，将多叉树根节点初始化
            if (NULL!= idl){
                SHGetPathFromIDList(idl,szBuffer);
                a[0].name=a[0].my_path=szBuffer;
                a[0].deep=0;
                flag=true;
            }
        }
        //判断是否手动输入了一个目录路径
        else if(_access(str.c_str(),0)!=-1){
                a[0].name=a[0].my_path=str;
                a[0].deep=0;
                flag=true;
        }
        //如果获取到有效路径，执行程序生成文件目录结构树，并输出。
        if(flag){
            build(0);
            printf("生成的文件目录树为：\n\n");
            print(0);
        }
        else
            printf("输入路径不存在或路径指向不为文件目录结构\n");
        printf("\n\n输入任意字符继续下一次生成文件目录结构\n");
//        getchar();
        system("pause");
    }
    return 0;
}