LeetCode 388. Longest Absolute File Path

原题链接在这里:https://leetcode.com/problems/longest-absolute-file-path/

题目:

Suppose we abstract our file system by a string in the following manner:

The string "dir\n\tsubdir1\n\tsubdir2\n\t\tfile.ext" represents:

dir
    subdir1
    subdir2
        file.ext

The directory dir contains an empty sub-directory subdir1 and a sub-directory subdir2 containing a file file.ext.

The string "dir\n\tsubdir1\n\t\tfile1.ext\n\t\tsubsubdir1\n\tsubdir2\n\t\tsubsubdir2\n\t\t\tfile2.ext" represents:

dir
    subdir1
        file1.ext
        subsubdir1
    subdir2
        subsubdir2
            file2.ext

The directory dir contains two sub-directories subdir1 and subdir2subdir1 contains a file file1.ext and an empty second-level sub-directory subsubdir1subdir2 contains a second-level sub-directory subsubdir2 containing a file file2.ext.

We are interested in finding the longest (number of characters) absolute path to a file within our file system. For example, in the second example above, the longest absolute path is "dir/subdir2/subsubdir2/file2.ext", and its length is 32 (not including the double quotes).

Given a string representing the file system in the above format, return the length of the longest absolute path to file in the abstracted file system. If there is no file in the system, return 0.

Note:

  • The name of a file contains at least a . and an extension.
  • The name of a directory or sub-directory will not contain a ..

Time complexity required: O(n) where n is the size of the input string.

Notice that a/aa/aaa/file1.txt is not the longest file path, if there is another path aaaaaaaaaaaaaaaaaaaaa/sth.png.

题解:

The longest absolute file path ends with file containing ".".

First separate input staring by "\n", now each line is one part.

For each line, have its last index of "\t" + 1 as level.

Now try to find its parent. While stack size is > level, pop the stack. When stack size == level, the stack top is the parent.

We have parent + "\" and append current part - length of "\t" as the new candidate.

If candidate contains ".", update the longest path length.

Push current candidate to the stack.

Time Complexity: O(n).n = input.length(). Each part could be in and out of stack once.

Space: O(n).

AC Java:

class Solution {
    public int lengthLongestPath(String input) {
        if(input == null || input.length() == 0){
            return 0;
        }
        
        String [] parts = input.split("\n");
        Stack<Integer> stk = new Stack<>();
        int res = 0;
        
        for(String part : parts){
            int level = part.lastIndexOf("\t") + 1;
            while(stk.size() > level){
                stk.pop();
            }
            
            int parent = stk.isEmpty() ? 0 : stk.peek() + 1;
            int can = parent + part.length() - level;
            if(part.indexOf(".") != -1){
                res = Math.max(res, can);
            }
            
            stk.push(can);
        }
        
        return res;
    }
}