【leetcode】删除子文件夹(字典树)

你是一位系统管理员，手里有一份文件夹列表 folder，你的任务是要删除该列表中的所有 子文件夹，并以 任意顺序 返回剩下的文件夹。

我们这样定义「子文件夹」：

如果文件夹 folder[i] 位于另一个文件夹 folder[j] 下，那么 folder[i] 就是 folder[j] 的子文件夹。
文件夹的「路径」是由一个或多个按以下格式串联形成的字符串：

/ 后跟一个或者多个小写英文字母。
例如，/leetcode 和 /leetcode/problems 都是有效的路径，而空字符串和 / 不是。

示例 1：

输入：folder = ["/a","/a/b","/c/d","/c/d/e","/c/f"]
输出：["/a","/c/d","/c/f"]
解释："/a/b/" 是 “/a” 的子文件夹，而 “/c/d/e” 是 “/c/d” 的子文件夹。
示例 2：

输入：folder = ["/a","/a/b/c","/a/b/d"]
输出：["/a"]
解释：文件夹 “/a/b/c” 和 “/a/b/d/” 都会被删除，因为它们都是 “/a” 的子文件夹。
示例 3：

输入：folder = ["/a/b/c","/a/b/d","/a/b/ca"]
输出：["/a/b/c","/a/b/ca","/a/b/d"]

提示：

1 <= folder.length <= 4 * 10^4
2 <= folder[i].length <= 100
folder[i] 只包含小写字母和 /
folder[i] 总是以字符 / 起始
每个文件夹名都是唯一的

思路分析:

一道判断前缀的问题，很容易想到构建字典树(前缀树)。但是直接构建，会发现/a/b/c和/a/b/ca不好判断。因为如果结束标记打在’c’上，那么/a/b/ca也算成/a/b/c了。因此我们需要在每个字符串后面多加一个’/'便于打结束标记。
同时还要注意先后顺序。如果先出现长的再出现短的，那么之前已经存了长的，就很难将其替换。因此我们要先对字符串数组进行排序。
下面上代码：

#define MAX 500020

class Solution {
public:
    int trie[MAX][28];	
    bool v[MAX];
    int k = 0;
    
    bool insert(string s)   
    {
        s += '/';   //多加一个'/'，便于划分前缀
        int rt = 0;
        int len = s.length();
        for(int i = 0;i < len;i++)
        {
            int id = s[i]-'a';
            if(s[i] == '/') id = 27;    //为'/'定一个节点编号
            if(v[rt]) return false;     //如果遍历中途发现结束标记，表示该前缀存在。
            if(!trie[rt][id])	
            {
                trie[rt][id] = ++k;
            }
            rt = trie[rt][id];
        }
        v[rt] = 1;	 
        return true;
    }

    
    vector<string> removeSubfolders(vector<string>& folder) {
        sort(folder.begin(), folder.end()); //先对字符串数组排序
        vector<string> ans;
        for(int i = 0;i < folder.size();i++)
        {
        	if(insert(folder[i]))
        	{
        		ans.push_back(folder[i]);
        	}
        }
		return ans;	
    }
};