二、刪除子文件夾(Weekly Contest 159)

題目描述:
你是一位系統管理員,手裏有一份文件夾列表 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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章