百练 2775：文件结构“图”（递归法）

2775:文件结构“图”

总时间限制: 

1000ms 

内存限制: 

65536kB

描述

在计算机上看到文件系统的结构通常很有用。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

提示

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

来源

翻译自 Pacific Northwest 1998 的试题

考虑到file要在dir输出完毕之后按照字母序输出，于是使用set来存储同一层中的文件名。遇到file时向set中insert文件名，遇到dir时则将文件层数增加一层，同时进行递归进入下一层，遇到]时先输出本层set中的文件名，然后将文件层数减去一层，return到上一层，遇到*时输出本层set中的文件名，同时return回main函数中，遇到#时将记录是否要继续的bool型变量stop置为true，同时return回main函数中。设置了几个全局变量来记录目前的数据组数、所处文件夹层数等。本题要求输出的"DATA SET x:"没有找到比较好的处理方式，只能又添加了一个print函数和若干全局变量来处理，若有更好的方式欢迎补充。下面附上源码：

#include <iostream>
#include <set>
using namespace std;
int n=0;//层数
int m=1;//第几组数据
bool have_print;
bool stop=false;
void print(void){
    if(have_print==false){
        have_print=true;
        if(m!=1) cout<<endl;
        cout<<"DATA SET "<<m<<":"<<endl<<"ROOT"<<endl;
    }
}
void fun(void){
    string str;
    set<string> s;
    while(cin>>str){
        switch (str[0]){
        case '#':
            stop=true;
            return;
        case 'f':
            print();
            s.insert(str);
            break;
        case 'd':
            print();
            n++;
            for(int i=0;i<n;i++){
                cout<<"|     ";
            }
            cout<<str<<endl;
            fun();
            break;
        case ']':
            for(set<string>::iterator i=s.begin();i!=s.end();i++){
                for(int j=0;j<n;j++){
                    cout<<"|     ";
                }
                cout<<*i<<endl;
            }
            n--;
            return;
        case '*':
            for(set<string>::iterator i=s.begin();i!=s.end();i++){
                cout<<*i<<endl;
            }
            m++;
            return;
        }
    }
}
int main(){
    while(!stop){
        have_print=false;
        fun();
    }
    return 0;
}