LeetCode 388.文件的最长绝对路径

题目链接

思路

针对文件路径的特征，一个文件中一定包含.分隔符，以此为依据可以判定当前字符串是否是一个文件，文件系统是一个树形结构的角度来看的话，题中给定的字符串实际上是以一个树形结构前序遍历的序列，连续的\t表示出了当前的深度，而相邻的节点之间以\n进行分割。
假设当前的路径为x/y/z，其中$x,y,z$的文件名长度为分别为$l_x,l_y,l_z$则路径&x, x/y, x/y/z&的长度分别为$l_x,l_x+l_y+1,l_x+l_y+l_z+2$。利用栈保存当前已遍历的路径长度，栈中元素个数即为当前路径的深度，栈顶元素即为当前路径的长度。设当前节点的文件名为q，当前节点的文件名长度为$l_p$，节点深度为depth，据此本题中应当可以有如下判断：

• 如果当前节点的深度大于当前路径的深度，则表明当前节点为栈顶节点的孩子节点，设当前栈顶结点的长度为t，栈顶节点的路径为p，则当前的文件路径应当为p/q
• 如果当前节点深度小于当前路径的深度，则表明当前节点不是栈顶节点的孩子节点，此时则需要一直退回到其父节点再进行求解。
• 由于题目只要求得出文件的长度，因此在实际运算中不需要保存具体的文件名称，只需要保存文件名的长度即可。检测当前节点的文件名的长度并标记当前的文件名是文件还是文件夹，如果当前的字符串为文件，则求出当前文件绝对路径的长度。遍历所有可能的文件长度，即可找到文件绝对路径的最大长度。

代码

class Solution {
public:
    int lengthLongestPath(string input) {
        int n = input.size();
        int pos = 0, ans = 0;
        stack<int> st;
 
        while(pos < n)
        {
            int depth = 1;
            while(pos < n && input[pos] == '\t')
            {
                depth++;
                pos++;
            }
            int len=0;
            bool isFile = false;
            while(pos < n && input[pos] != '\n')
            {
                if(input[pos] == '.')   isFile = true;
                pos++;
                len++;
            }
            pos++;//换行符
            while(st.size() >= depth)   st.pop();
            if(!st.empty()) len+=st.top() + 1;
            if(isFile)  ans = max(ans, len);
            else st.emplace(len);  
        }
        return ans;
    }
};