OpenJudge(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个空格来表示出缩排的层次。

看似题目很长，但是就是一道模拟题~~

但是我一开始卡住了就是因为不知道要怎么样在一个个子目录中保存一个个的文件，这样子的话char好像要开三维？？所以后来学习了用栈+stl的vector+string来搞。就是当遇到一个dir时，然后把'['这个字符压入到栈中，然后去所有的字符中寻找']'这个字符，如果找到了，则把这个字符之间的所有在栈中的字符都弹出来，然后保存到vector中去，然后sort一遍，然后输出。其他的应该没有什么难点了~~~

#include<cstdio>
#include<cstring>
#include<set>
#include<cmath>
#include<map>
#include<stack>
#include<string>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 44
typedef long long ll;
string ss;
stack<string> idd;
vector<string> akk;
int vis[maxn*maxn];
struct node{
    int f;  //f=0 :file ; f=1: dir;
    string tmp;
    int weizhi;
}a[maxn*maxn];
int main(){
    int num=0,jj=1;
    while(cin>>ss){
        int tot=0; num=0;
        if(ss[0]=='#') break;
        memset(vis,0,sizeof(vis));
        if(ss[0]=='f'){
            a[num].f=0;  a[num].tmp=ss;
            a[num++].weizhi=tot;
        }
        else if(ss[0]=='d'){
            tot++;
            a[num].f=1; a[num].tmp=ss;
            a[num++].weizhi=tot;
        }
        while(cin>>ss){
            if(ss[0]=='*') break;
            if(ss[0]==']'){
                a[num].f=-2; a[num].tmp=ss;
                num++;
                tot--; continue;
            }
            else if(ss[0]=='f'){
                a[num].f=0;  a[num].tmp=ss;
                a[num++].weizhi=tot;
            }
            else if(ss[0]=='d'){
                tot++;
                a[num].f=1;  a[num].tmp=ss;
                a[num++].weizhi=tot;
            }
        }
        printf("DATA SET %d:\nROOT\n",jj++);
        tot=0;  memset(vis,0,sizeof(vis));
        for(int i=0;i<num;i++){
            if(vis[i]) continue;
            if(a[i].f==1){
                vis[i]=1;
                tot++;
                if(a[i].weizhi==tot){
                    for(int j=1;j<=tot;j++) printf("|     ");
                    cout<<a[i].tmp<<endl;
                    idd.push("[");
                    //接下来是要曲去找与之匹配的结尾了;
                    int ff=0;
                    for(int j=i+1;j<num;j++){
                        if(ff==-1) break;
                        if(a[j].tmp[0]==']'){
                            vis[j]=1;
                            ff--;
                            akk.clear();
                            while(!idd.empty()){
                                string tmp=idd.top();
                                idd.pop();
                                if(tmp[0]=='[') break;
                                akk.push_back(tmp);
                            }
                            sort(akk.begin(),akk.end());
                            int len=akk.size();
                            for(int kk=0;kk<len;kk++){
                                for(int k=1;k<=tot;k++) printf("|     ");
                                cout<<akk[kk]<<endl;
                            }
                            tot--;
                            if(ff==-1) break;
                        }
                        else if(a[j].f==1){     //如果是dir
                            vis[j]=1;
                            tot++;
                            if(a[j].weizhi==tot){
                                for(int k=1;k<=tot;k++) printf("|     ");
                                cout<<a[j].tmp<<endl;
                                ff++;
                                idd.push("[");
                            }
                        }
                        else if(a[j].f==0){
                            vis[j]=1;
                            idd.push(a[j].tmp);
                        }
                    }
                }
            }
            else idd.push(a[i].tmp);
        }
        akk.clear();
        while(!idd.empty()){
            string tmp=idd.top();
            idd.pop();
            akk.push_back(tmp);
        }
        sort(akk.begin(),akk.end());
        int len=akk.size();
        for(int i=1;i<=tot;i++) printf("|     ");
        for(int i=0;i<len;i++){
            cout<<akk[i]<<endl;
        }
        puts("");
    }
    return 0;
}
/*
file1
file2
dir3
dir2
file1
file2
]
]
file4
dir1
]
file3
*
file2
file1
*
#
*/