OPENJUDGE 2775 文件结构图

时间限制: 

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

提示

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

解题思路：

用栈记录输入项，用优先队列排序文件，当遇到‘]’时就弹栈直至遇见第一个dir，并将所有的file加入priority queue,排序完成后输出，利用LevelCount记录缩进次数，遇到dir缩进+1，遇到‘]’-1. 用setBegin记录是否为新文件集，若是则输出开头信息

# include <iostream>
# include <string>
# include <stack>
# include <set>
# include <queue>

using namespace std;

void printIndentation(int n)
{
	for ( int i = 0; i < n; i++ )
	{
		cout <<"|     ";
	}
}

int main()
{
	int setNum = 1;
	int levelCount = 0;
	bool setBegin = true;
	string inputString;
	string currentString;
	stack<string> inputStack;
	priority_queue<string,vector<string>,greater<string> > sorter;


	while ( cin >> inputString )
	{
		currentString = "";
		if ( inputString == "#" )
		{
			break;
		}
		if ( inputString == "*" )
		{
			while ( !inputStack.empty() )
			{
				currentString = inputStack.top();
				sorter.push(currentString);
				inputStack.pop();
			}
			while ( !sorter.empty() )
			{
				currentString = sorter.top();
				printIndentation(levelCount);
				cout << currentString << endl;
				sorter.pop();
			}
			cout << endl;
			setBegin = true;
			levelCount = 0;
			setNum++;
			continue;
		}
		if ( setBegin == true )
		{
			cout << "DATA SET " << setNum << ":" << endl;
			cout << "ROOT" << endl;
			setBegin = false;
		}
		if ( inputString[0] == 'd' )
		{
			levelCount++;
			printIndentation(levelCount);
			cout << inputString << endl;
			inputStack.push(inputString);
		}
		if ( inputString == "]" )
		{
			while ( !inputStack.empty() && currentString[0] != 'd' )		//pop files out until the first dir occurs
			{
				currentString = inputStack.top();	
				if ( currentString[0] == 'f' )
				{
					sorter.push(currentString);
				}
				inputStack.pop();
			}
			while ( !sorter.empty() )
			{
				currentString = sorter.top();
				printIndentation(levelCount);
				cout << currentString << endl;
				sorter.pop();
			}
			levelCount--;
		}
		if ( inputString[0] == 'f' )
		{
			inputStack.push(inputString);
		}
	}
	return 0;
}