【Leetcode】588. Design In-Memory File System

题目地址：

https://leetcode.com/problems/design-in-memory-file-system/description/

要求实现一个文件系统的数据结构，可以做如下操作：
1、"ls"：给定一个路径$p$，$p$的格式为'/'后面跟文件名跟'/'等等，和普通Linux系统一样，如果这个路径指向一个文件，则返回这个文件名；如果指向一个文件夹，则返回该文件夹中所有文件和文件夹的名字，并按字典序排序。
2、"mkdir"：给定一个路径，沿着这个路径创建文件夹；
3、给定一个路径，将某个内容加入这个路径指向的文件；
4、给定一个路径，查询这个路径指向的文件的内容。
题目保证所有文件、文件夹的名字都是英文小写，同一个文件夹下不会存在名字相同的两个元素，并且每次询问保证一定有效。

文件系统大致是一个树结构，由于一个路径可能是文件也可能是文件夹，我们开个struct来存这个路径的文件夹的名字，然后用两个哈希表，一个存其所含的所有文件夹的节点，另一个存其所含的所有文件的内容。所有操作都是类似的，即按照"/"做split，然后一层一层向下走，走到最后一个分隔符的时候，要看一下最后的那个名字是文件还是文件夹，对应做操作即可。代码如下：

class FileSystem {
 public:
  struct Dir {
    unordered_map<string, Dir *> dirs;
    unordered_map<string, string> files;
  };

  Dir *root;

  FileSystem() { root = new Dir(); }

  vector<string> ls(string path) {
    auto *f = root;
    string fname;
    for (int i = 1; i < path.size(); i++) {
      int j = i;
      while (j < path.size() && path[j] != '/') j++;
      fname = path.substr(i, j - i);
      if (j == path.size()) break;
      f = f->dirs[fname];
      i = j;
    }
	
	// 如果fname不空的话，我们看一下其是文件还是文件夹
	// 如果是文件，则直接返回之；如果是文件夹，则向下走一层
    if (fname.size()) {
      if (f->files.count(fname)) return {fname};
      f = f->dirs[fname];
    }
    vector<string> res;
    for (auto &[k, v] : f->dirs) res.push_back(k);
    for (auto &[k, v] : f->files) res.push_back(k);
    sort(res.begin(), res.end());
    return res;
  }

  void mkdir(string path) {
    auto *f = root;
    for (int i = 1; i < path.size(); i++) {
      int j = i;
      while (j < path.size() && path[j] != '/') j++;
      auto fname = path.substr(i, j - i);
      if (!f->dirs.count(fname)) f->dirs[fname] = new Dir();
      f = f->dirs[fname];
      i = j;
    }
  }

  void addContentToFile(string path, string content) {
    auto *f = root;
    string fname;
    for (int i = 1; i < path.size(); i++) {
      int j = i;
      while (j < path.size() && path[j] != '/') j++;
      fname = path.substr(i, j - i);
      if (j == path.size()) break;
      f = f->dirs[fname];
      i = j;
    }
    f->files[fname] += content;
  }

  string readContentFromFile(string path) {
    auto *f = root;
    string fname;
    for (int i = 1; i < path.size(); i++) {
      int j = i;
      while (j < path.size() && path[j] != '/') j++;
      fname = path.substr(i, j - i);
      if (j == path.size()) break;
      f = f->dirs[fname];
      i = j;
    }
    return f->files[fname];
  }
};

时空复杂度取决于具体数据，每次询问时间大致与路径长度成正比。