百练2775:文件结构“图”

描述

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

#include <cstdio>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;
void print(int level){
    for(int i = 0;i < level;i++)
        printf("|     ");
}
int main(){
    char temp[32];
    int cnt = 1;
    vector<string> v;
    vector<string> content[100];
    while(true){
        v.clear();
        scanf("%s",temp);
        if(temp[0] == '#')break;
        else{
            string s = temp;
            v.push_back(s);
        }
        if(cnt == 1)
            printf("DATA SET %d:
ROOT
",cnt++);
        else
            printf("
DATA SET %d:
ROOT
",cnt++);
        while(~scanf("%s",temp)){
            if(temp[0] == '*')break;
            string t = temp;
            v.push_back(t);
        }
        //
        int size = v.size();
        int level = 0;
        content[0].clear();
        for(int i = 0;i < size;){
            if(v[i][0] == 'd'){
                level++;
                content[level].clear();
                print(level);
                printf("%s
",v[i].c_str());
                if(i < size-1 && v[i+1][0] == ']'){
                    level--;
                    i += 2;
                }else{
                    //level++;
                    i++;
                }
            }else if(v[i][0] == 'f'){
                content[level].push_back(v[i]);
                i++;
            }else if(v[i][0] == ']'){
                sort(content[level].begin(),content[level].end());
                for(int p = 0;p < content[level].size();p++){
                    print(level);printf("%s
",content[level][p].c_str());
                }
                level--;
                i++;
            }
        }
        sort(content[level].begin(),content[level].end());
        for(int i = 0;i < content[level].size();i++){
            print(level);printf("%s
",content[level][i].c_str());
        }
    }

}