编程题＃2： 文件结构“图”

编程题＃2： 文件结构“图”

来源: POJ (Coursera声明：在POJ上完成的习题将不会计入Coursera的最后成绩。)

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

描述

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

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

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

输入

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

输出

在显示一个目录中内容的时候，先显示其中的子目录（如果有的话），然后再显示文件（如果有的话）。文件要求按照名字的字母表的顺序显示（目录不用按照名字的字母表顺序显示，只需要按照目录出现的先后显示）。对每一组测试数据，我们要先输出"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

提示

一个目录和它的子目录处于不同的层次

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

代码

可通过OJ和Coursera的代码请参考文件结构“图”。
以下代码为我个人修改了输入输出后的内容，题目中的样例没问题（可惜不知道为啥WA）

#include <iostream>
#include <string>
#include <queue>

using namespace std;

int judge(string str[],int n)
{
    int count_lev=1;
    priority_queue <string,vector<string>, greater<string> > que_file[1000];
    for(int i=0;i<n;i++) {
        if (str[i][0]=='d') {
            for(int j=0;j<count_lev;j++) {
                cout << "|     ";
            }
            cout << str[i] << endl;
            count_lev++;
        }
        if (str[i][0]=='f') {
            que_file[count_lev].push(str[i]);
        }
        if (str[i][0]==']' || i==n-1) {
            while(!que_file[count_lev].empty()) {
                for (int j=0;j< count_lev-1;j++)  {
                    cout << "|     ";
                }
                cout << que_file[count_lev].top() <<endl;
                que_file[count_lev].pop();
            }
            count_lev--;
        }
        if (i==n-1) cout << endl;
    }
    return 0;
}

int main()
{
    string all_info[1000];
    string each_info;
    int count_all=0;
    
    // 将输入内容全部存入字符串数组
    while(cin >> each_info) {
        if(each_info[0]=='#') break;
        else {
            all_info[count_all]=each_info;
            count_all++;
        }
    }
    
    string set_info[1000];
    int j=0;
    int k=1;
    
    // 将没一个部分的内容分别拿出来单独进行处理
    for(int i=0;i<=count_all;i++) {
        if(all_info[i][0]!='*') {
            set_info[j]=all_info[i];
            j++;
        }
        else {
            cout << "DATA SET " << k << ":" <<endl;
            cout << "ROOT" <<endl;
            judge(set_info,j);
            j=0;
            k++;
        }
    }
    return 0;
}