WEEK9-A-咕咕东的目录管理器

题面
咕咕东的雪梨电脑的操作系统在上个月受到宇宙射线的影响，时不时发生故障，他受不了了，想要写一个高效易用零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;
}