PAT4-08. 目录树

#include<string>
#include<iostream>
#include<map>
using namespace std;
struct dir{map<string,dir*>content;};

void add(dir*p,string s){
  if(s==string())return;
  auto i=s.find('\\');
  string nname=s.substr(0,i);
  s.erase(0,i+1);
  if(i==string::npos){
    p->content[nname]=nullptr;
  }else{
    if(p->content.count(nname)==0)
      p->content[nname]=new dir;
    add(p->content[nname],s);
  }//else
}
// 'print' print dirs and files in dir p
void print(dir*p,int deep){
  for(auto x:p->content)
    if(x.second!=nullptr){
      cout<<string(deep<<1,' ')<<x.first<<endl;
      print(x.second,deep+1);
    } // for
  for(auto x:p->content)
    if(x.second==nullptr)
      cout<<string(deep<<1,' ')<<x.first<<endl;
}

int main(){
  dir*root=new dir;
  int n;cin>>n;
  while(n--){
    string s;cin>>s;
    add(root,s);    
  }
  cout<<"root"<<endl;
  print(root,1);
  return 0;
}

读取字符串时，名字后边有个‘\'的就是文件夹，没有的就是普通文件，文件夹中的内容存在一个map<string,dir*>中，普通文件的dir*为nullptr，以此区分文件夹中的子文件夹和普通文件。