算法基础---递归题目 之 文件结构“图”

时间及空间要求：

总时间限制: 1000ms 内存限制: 65536kB

问题描述(问题部分转自：http://cxsjsxmooc.openjudge.cn/test2/B/)：

在计算机上看到文件系统的结构通常很有用。Microsoft Windows上面的”explorer”程序就是这样的一个例子。但是在有图形界面之前，没有图形化的表示方法的，那时候最好的方式是把目录和文件的结构显示成一个”图”的样子，而且使用缩排的形式来表示目录的结构。比如：

ROOT
|     dir1
|     file1
|     file2
|     file3
|     dir2
|     dir3
|     file1
file1
file2

这个图说明：ROOT目录包括三个子目录和两个文件。第一个子目录包含3个文件，第二个子目录是空的，第三个子目录包含一个文件。

程序输入：

写一个程序读取一些测试数据。每组测试数据表示一个计算机的文件结构。每组测试数据以’*’结尾，而所有合理的输入数据以’#’结尾。一组测试数据包括一些文件和目录的名字（虽然在输入中我们没有给出，但是我们总假设ROOT目录是最外层的目录）。在输入中,以’]’表示一个目录的内容的结束。目录名字的第一个字母是’d’，文件名字的第一个字母是’f’。文件名可能有扩展名也可能没有（比如fmyfile.dat和fmyfile）。文件和目录的名字中都不包括空格,长度都不超过30。一个目录下的子目录个数和文件个数之和不超过30。

程序输出：

在显示一个目录中内容的时候，先显示其中的子目录（如果有的话），然后再显示文件（如果有的话）。文件要求按照名字的字母表的顺序显示（目录不用按照名字的字母表顺序显示，只需要按照目录出现的先后显示）。对每一组测试数据，我们要先输出”DATA SET x:”，这里x是测试数据的编号（从1开始）。在两组测试数据之间要输出一个空行来隔开。

需要注意的是，我们使用一个’|’和5个空格来表示出缩排的层次。

样例输入:

file1
file2
dir3
dir2
file1
file2
]
]
file4
dir1
]
file3
*
file2
file1
*
#

样例输出:

DATA SET 1:
ROOT
|     dir3
|     |     dir2
|     |     file1
|     |     file2
|     dir1
file1
file2
file3
file4

DATA SET 2:
ROOT
file1
file2

提示:

一个目录和它的子目录处于不同的层次
一个目录和它的里面的文件处于同一层次

代码及解决思路：

/*
解决思路：采用递归方法，初始化建立root文件夹，从输入流中读取字符串，若遇到文件（file），则将其insert
到文件集合（采用STL中set容器，能够自动对文件名进行排序）中，若读取到文件夹（dir），则新建一级文件夹，
并调用文件夹类方法进行构造下级相关文件结构，若读取到文件夹结束符（']'）或者文件结束符（‘*’），则将
return退出当前子函数，回到上一层次的函数之中。问题相对较为简单，使用STL容器时尽量使用指针，能够避免很
多不必要的麻烦（例如对象的深度拷贝等）。
*/

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;

class File {
protected:
    char *file_name_;
public:
    File(const char *name){
        file_name_ = new char[strlen(name)+1];
        memset(file_name_,0,strlen(name)+1);
        memcpy(file_name_,name,strlen(name)+1);
    }
    ~File(){
        if (file_name_ == NULL )
            delete []file_name_;
        file_name_ = NULL;
    }

    void Print(int depth) const {
        for (int i = 0 ; i < depth - 1 ; i++)
            cout << "|     " ;
        cout << file_name_ << endl;
    }
    friend class SortByWhat;
};

class SortByWhat{
public:
    bool operator()(const File *f1,const File *f2){
        if(strcmp(f1->file_name_,f2->file_name_) < 0 )
            return true;
        else
            return false;
    }
};

class Dir {
private:
    char *dir_name_;
    vector<Dir*> dir_vec_;              //文件夹
    set<File*,SortByWhat> file_vec_;    //文件
public:
    Dir(const char *name){
        dir_name_ = new char[strlen(name)+1];
        memcpy(dir_name_,name,strlen(name)+1);
    }
    ~Dir(){
        delete []dir_name_;
    }
    void ReadFromStream(){
        char name[255]={0};
        while(cin >> name){
            switch(name[0]){
                case 'f':{
                    File *f = new File(name);
                    file_vec_.insert(f);
                    break;
                }
                case 'd':{
                    Dir *d =new Dir(name);
                    dir_vec_.push_back(d);
                    d->ReadFromStream();
                    break;
                }
                case ']':break;
                case '*':break;
                case '#':{
                    char a = '#';
                    throw a;
                    break;
                }
                default:break;
            }
            if (name[0] == '*' || name[0] == ']') break;
        }
    }

    void Print(int depth) const {
        for (int i = 0 ; i < depth ; i++)
            cout << "|     " ;
        cout << dir_name_ << endl;
        for (auto iter = dir_vec_.begin();iter != dir_vec_.end(); iter ++){
            (*iter)->Print(depth+1);
        }
        for (set<File*,SortByWhat>::iterator iter = file_vec_.begin();iter != file_vec_.end(); iter++){
            (*iter)->Print(depth+1);
        }
    }
};


int main()
{
    int set_case = 1;
    try{
        while(true){
            Dir root("ROOT");
            root.ReadFromStream();
            printf("DATA SET %d:\n",set_case);
            root.Print(0);
        }
    } catch(char e) {
        // 这里采用异常的方式退出有点不和谐。
    }catch(...){
        cerr << "Something unexpacted get." << endl;
        return 1;
    }
    return 0；
}