題目描述:
你是一位系統管理員,手裏有一份文件夾列表 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] 總是以字符 / 起始
每個文件夾名都是唯一的
超時的代碼
class Solution {
public List<String> removeSubfolders(String[] folder) {
Arrays.sort(folder);
Set<Integer> set = new HashSet<>();
for (int i = 0; i < folder.length; i++) {
String s = folder[i];
for (int j = 0; j < folder.length && j != i; j++) {
// System.out.println(folder[j]);
if(s.charAt(0) != folder[j].charAt(0) )
break;
if(s.startsWith(folder[j]) && s.charAt(folder[j].length()) == '/'){
set.add(i);
break;
}
}
}
List<String> list = new ArrayList<>();
for (int i = 0; i < folder.length; i++) {
String s = folder[i];
if(!set.contains(i)){
list.add(s);
}
}
return list;
}
}
沒想到合適的思路,前綴樹????
看看別人的思路:
按照長度從小到大排序,然後就保證不會遺漏啦~~~,我真是chun。。。
class Solution {
public List<String> removeSubfolders(String[] folder) {
int n = folder.length;
List<String> res;
Arrays.sort(folder, (a1,a2)->a1.length() - a2.length());
Set<String> set = new HashSet<>();
for (int i = 0; i < n; i++) {
String s = folder[i];
if (s.length() <= 1) {
continue;
}
boolean flag = false;
for (int j = 1; j < s.length(); j++) {
if ((j == s.length() - 1 || s.charAt(j) == '/') && set.contains(s.substring(0, j + (j == s.length() - 1 ? 1 : 0)))) {
flag = true;
break;
}
}
if (!flag) {
set.add(s);
}
}
res = new ArrayList<>(set);
return res;
}
}