POJ 1057 FILE MAPPING(文件结构图)

描述

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

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

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

输入

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

输出

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

样例输入

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

分析

既然是存文件信息，用递归模仿相关File类保存好文件结构即好，输出按想输出的方式输出即可。

实现

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
using std::string;
using std::vector;
const string ROOT_NAME="ROOT";
const string SPACE_UNIT="|     ";

struct File{
    string name;
    vector<File> dirs;
    vector<string> files;
};

//1:* 2:#
int buildStructure(File *file){
    string name;
    while (std::cin >> name) {
        if (name == "*") {
            sort(file->files.begin(), file->files.end());
            return 1;
        }
        if (name == "#") {
            sort(file->files.begin(), file->files.end());
            return 2;
        }
        if (name == "]"){
            sort(file->files.begin(), file->files.end());
            return 0;
        }
        if (name[0] == 'f'){
            file->files.push_back(name);
        }else{
            File subFile;
            subFile.name = name;
            int result = buildStructure(&subFile);
            file->dirs.push_back(subFile);
            if (result) {
                return result;
            }
        }
    }
    return 0;
}

void print(File *dir, string spaces){
    string subSpaces = spaces + SPACE_UNIT;
    std::cout << spaces << dir->name << std::endl;

    for(vector<File>::iterator iter = dir->dirs.begin(); iter != dir->dirs.end(); ++iter){
        print(&(*iter), subSpaces);
    }

    for(vector<string>::iterator iter = dir->files.begin(); iter != dir->files.end(); ++iter){
        std::cout << spaces << *iter << std::endl;
    }
}

int main(){
//    freopen("in.txt", "r", stdin);
    int cases = 0;
    string name;
    while(1){
        File root;
        root.name = ROOT_NAME;
        int result = buildStructure(&root);
        if (result == 2) {
            break;
        }
        if (cases > 0) {
            std::cout << std::endl;
        }
        std::cout << "DATA SET " << ++cases << ":" << std::endl;
        print(&root, "");
    }
    return 0;
}