题面
咕咕东的雪梨电脑的操作系统在上个月受到宇宙射线的影响,时不时发生故障,他受不了了,想要写一个高效易用零bug的操作系统 —— 这工程量太大了,所以他定了一个小目标,从实现一个目录管理器开始。前些日子,东东的电脑终于因为过度收到宇宙射线的影响而宕机,无法写代码。他的好友TT正忙着在B站看猫片,另一位好友瑞神正忙着打守望先锋。现在只有你能帮助东东!
初始时,咕咕东的硬盘是空的,命令行的当前目录为根目录 root。
目录管理器可以理解为要维护一棵有根树结构,每个目录的儿子必须保持字典序。
现在咕咕东可以在命令行下执行以下表格中描述的命令:
命令
类型
实现
说明
MKDIR s
操作
在当前目录下创建一个子目录 s,s 是一个字符串
创建成功输出 “OK”;若当前目录下已有该子目录则输出 “ERR”
RM s
操作
在当前目录下删除子目录 s,s 是一个字符串
删除成功输出 “OK”;若当前目录下该子目录不存在则输出 “ERR”
CD s
操作
进入一个子目录 s,s 是一个字符串(执行后,当前目录可能会改变)
进入成功输出 “OK”;若当前目录下该子目录不存在则输出 “ERR”
特殊地,若 s 等于 “…” 则表示返回上级目录,同理,返回成功输出 “OK”,返回失败(当前目录已是根目录没有上级目录)则输出 “ERR”
SZ
询问
输出当前目录的大小
也即输出 1+当前目录的子目录数
LS
询问
输出多行表示当前目录的 “直接子目录” 名
若没有子目录,则输出 “EMPTY”;若子目录数属于 [1,10] 则全部输出;若子目录数大于 10,则输出前 5 个,再输出一行 “…”,输出后 5 个。
TREE
询问
输出多行表示以当前目录为根的子树的前序遍历结果
若没有后代目录,则输出 “EMPTY”;若后代目录数+1(当前目录)属于 [1,10] 则全部输出;若后代目录数+1(当前目录)大于 10,则输出前 5 个,再输出一行 “…”,输出后 5 个。若目录结构如上图,当前目录为 “root” 执行结果如下,
UNDO
特殊
撤销操作
撤销最近一个 “成功执行” 的操作(即MKDIR或RM或CD)的影响,撤销成功输出 “OK” 失败或者没有操作用于撤销则输出 “ERR”
输入
输入文件包含多组测试数据,第一行输入一个整数表示测试数据的组数 T (T <= 20);
每组测试数据的第一行输入一个整数表示该组测试数据的命令总数 Q (Q <= 1e5);
每组测试数据的 2 ~ Q+1 行为具体的操作 (MKDIR、RM 操作总数不超过 5000);
面对数据范围你要思考的是他们代表的 “命令” 执行的最大可接受复杂度,只有这样你才能知道你需要设计的是怎样复杂度的系统。
输出
每组测试数据的输出结果间需要输出一行空行。注意大小写敏感。
时空限制
Time limit 6000 ms
Memory limit 1048576 kB
样例输入
1
22
MKDIR dira
CD dirb
CD dira
MKDIR a
MKDIR b
MKDIR c
CD …
MKDIR dirb
CD dirb
MKDIR x
CD …
MKDIR dirc
CD dirc
MKDIR y
CD …
SZ
LS
TREE
RM dira
TREE
UNDO
TREE
样例输出
OK
ERR
OK
OK
OK
OK
OK
OK
OK
OK
OK
OK
OK
OK
OK
9
dira
dirb
dirc
root
dira
a
b
c
dirb
x
dirc
y
OK
root
dirb
x
dirc
y
OK
root
dira
a
b
c
dirb
x
dirc
y
思路
本题如果没有TREE和UNDO操作的话,就是一个简单的树的问题。由于这两个操作的出现我们需要加入一些量。
先说一下UNDO操作:
由于需要回溯最后一步带来的影响,所以我们需要用一个数组来记录每次进行的操作,所以我们用一个vector数组来记录 进行的操作。遇到UNDO时弹出最后一个,根据MKDIR和RM和CD进行恢复的操作。要注意的是如果是MKDIR和RM的时候别忘了从当前目录开始到根目录进行update操作,即把改变的siz复原。
然后是最难的TREE操作:
首先要明确,不可能每一次TREE操作都要遍历一次,否则的话复杂度最高有(e5-5000)*5000,会超时。所以肯定会有重复的TREE操作。所以我们需要设计一个变量tag,以保存是否被访问过,这样就可以解决重复问题了。注意的是我们也要在update操作里面加入改变tag为0的操作,因为很可能已经改变了遍历的结果,所以要重新遍历。遍历用到迭代器和递归,值得注意的是迭代器返回的end()是不能用的,也就是迭代器左闭右开,一定要注意,否则会MLE。
代码
#include<iostream>
#include<string>
#include<string.h>
#include<vector>
#include<map>
using namespace std;
const string str[]={"MKDIR","RM","CD","SZ","LS","TREE","UNDO"};
int cur;//当前目录
int num;
struct Node{
string name;
int siz;
int father;
map<string,int> mp;
vector<string> pre,bck;
bool tag;
}node[5010];
struct ope{
string name;
int idx;
ope(string n,int i){
name=n;
idx=i;
}
};
vector<ope> op;
void init(){
cur=0,num=1;
node[0].father=-1;
node[0].name="root";
node[0].siz=1;
node[0].tag=false;
node[0].pre.clear();
node[0].bck.clear();
node[0].mp.clear();
op.clear();
}
void create(string s){
node[cur].mp[s]=num;
node[num].name=s;
node[num].siz=1;
node[num].father=cur;
node[num].mp.clear();
node[num].pre.clear();
node[num].bck.clear();
node[num].tag=false;
}
void MKDIR(string s){
if(node[cur].mp.count(s)){
cout<<"ERR"<<endl;
}
else{
create(s);
int i=cur;
while(i!=-1){
node[i].tag=false;
node[i].siz++;
i=node[i].father;
}
ope opr(str[0],num);
op.push_back(opr);
num++;
cout<<"OK"<<endl;
}
}
void RM(string s){
if(!node[cur].mp.count(s))
cout<<"ERR"<<endl;
else{
node[cur].tag=false;
int i=node[cur].mp[s];
int sz=node[i].siz;
node[cur].mp.erase(s);
int j=cur;
while(j!=-1){
node[j].tag=false;
node[j].siz-=sz;
j=node[j].father;
}
ope opr(str[1],i);
op.push_back(opr);
cout<<"OK"<<endl;
}
}
void CD(string s){
if(s==".."){
if(node[cur].name=="root")
cout<<"ERR"<<endl;
else{
ope opr(str[2],cur);
op.push_back(opr);
cur=node[cur].father;
cout<<"OK"<<endl;
}
}
else{
if(!node[cur].mp.count(s))
cout<<"ERR"<<endl;
else{
ope opr(str[2],cur);
op.push_back(opr);
int i=node[cur].mp[s];
cur=i;
cout<<"OK"<<endl;
}
}
}
void SZ(){
cout<<node[cur].siz<<endl;
}
void LS(){
if(node[cur].mp.empty())
cout<<"EMPTY"<<endl;
else{
if(node[cur].mp.size()<=10){
for(map<string,int>::iterator it=node[cur].mp.begin();it!=node[cur].mp.end();it++){
cout<<it->first<<endl;
}
}
else{
map<string,int>::iterator it=node[cur].mp.begin();
for(int i=1;i<=5;i++)
{
cout<<it->first<<endl;
it++;
}
cout<<"..."<<endl;
it=node[cur].mp.end();
for(int i=1; i<=5; i++)
it--;
for(int i=1; i<=5; i++)
{
cout<<it->first<<endl;
it++;
}
}
}
}
void preorder(int id,vector<string> &pre){
pre.push_back(node[id].name);
for(map<string,int>::iterator it=node[id].mp.begin(); it!=node[id].mp.end(); it++){
preorder(it->second,pre);
}
}
void pretrack(int id,int &cnt,vector<string> &pre){
pre.push_back(node[id].name);
cnt--;
if(cnt==0) return;
for(map<string,int>::iterator it=node[id].mp.begin(); it!=node[id].mp.end(); it++){
pretrack(it->second,cnt,pre);
if(cnt==0) return;
}
}
void bcktrack(int id,int &cnt,vector<string> &bck){
map<string,int>::iterator it=node[id].mp.end();
while(it!=node[id].mp.begin())
{
it--;
bcktrack(it->second,cnt,bck);
if(cnt==0) return;
}
bck.push_back(node[id].name);
cnt--;
}
void pushdown(int id){
node[id].pre.clear();
node[id].bck.clear();
if(node[id].siz<=10){
preorder(id,node[id].pre);
}
else{
int cnt=5;
pretrack(id,cnt,node[id].pre);
cnt=5;
bcktrack(id,cnt,node[id].bck);
}
node[id].tag=true;
}
void TREE(){
if(!node[cur].tag)
pushdown(cur);
if(node[cur].siz==1)
cout<<"EMPTY"<<endl;
else {
if(node[cur].siz<=10)
{
//cout<<node[cur].pre.size()<<endl;
for(int i=0;i<node[cur].pre.size();i++)
cout<<node[cur].pre[i]<<endl;
}
else
{
for(int i=0;i<5;i++)
cout<<node[cur].pre[i]<<endl;
cout<<"..."<<endl;
for(int i=4;i>=0;i--)
cout<<node[cur].bck[i]<<endl;
}
}
}
void UNDO(){
if(op.empty())
cout<<"ERR"<<endl;
else{
ope opr=op[op.size()-1];
op.pop_back();
if(opr.name==str[0])
{
int i=opr.idx;
int sz=node[i].siz;
node[cur].mp.erase(node[i].name);
int j=cur;
while(j!=-1){
node[j].tag=false;
node[j].siz-=sz;
j=node[j].father;
}
//num--;
}
else if(opr.name==str[1])
{
int i=opr.idx;
int sz=node[i].siz;
node[cur].mp[node[i].name]=i;
int j=cur;
while(j!=-1){
node[j].tag=false;
node[j].siz+=sz;;
j=node[j].father;
}
}
else
cur=opr.idx;
cout<<"OK"<<endl;
}
}
int main(){
int T;
cin>>T;
for(int i=0;i<T;i++)
{
int Q;
cin>>Q;
init();
for(int j=0;j<Q;j++)
{
string opp;
cin>>opp;
if(opp==str[0])
{
string s;
cin>>s;
MKDIR(s);
}
else if(opp==str[1])
{
string s;
cin>>s;
RM(s);
}
else if(opp==str[2])
{
string s;
cin>>s;
CD(s);
}
else if(opp==str[3])
SZ();
else if(opp==str[4])
LS();
else if(opp==str[5])
TREE();
else if(opp==str[6])
UNDO();
}
}
return 0;
}