588_Design_In_Memory_File_System

Author: Partho Biswas

README.md

@@ -489,6 +489,7 @@ Check this [golden](https://tinyurl.com/ujopecz) post.
 |02| [211. Add and Search Word. Data structure design](https://leetcode.com/problems/add-and-search-word-data-structure-design/)| [Python](https://tinyurl.com/wu6rdaw/211_Add_and_Search_Word_Data_structure_design.py)| [Video 01](https://www.youtube.com/watch?v=qi2ohSEyyDw), [Video 02](https://www.youtube.com/watch?v=neb_2UK5Kuo&t=13s)|
 |03| [642. Design Search Autocomplete System](https://leetcode.com/problems/design-search-autocomplete-system/)| [Python](https://tinyurl.com/wu6rdaw/642_Design_Search_Autocomplete_System.py)| [Article 01](https://leetcode.com/problems/design-search-autocomplete-system/discuss/105386/Python-Clean-Solution-Using-Trie), [Video 01](https://www.youtube.com/watch?v=us0qySiUsGU)| Hard | --- |
 |04| [472. Concatenated Words](https://tinyurl.com/y9cv6dk5)| [Python](https://tinyurl.com/wu6rdaw/472_Concatenated_Words.py)| [Art 01](https://tinyurl.com/ycjm6fuo), [Art 02](https://tinyurl.com/ybvl3kbv)| Hard | Very important |
+|05| **[588. Design In-Memory File System](https://tinyurl.com/ya238cp4)** | [Python](https://tinyurl.com/wu6rdaw/588_Design_In_Memory_File_System.py)| --- | Hard | --- |
 
 </p>
 </details>

leetcode.com/python/588_Design_In_Memory_File_System.py

@@ -0,0 +1,91 @@
+class TrieNode:
+    def __init__(self):
+        self.children = {}
+        self.childDirSet = set()
+        self.name = None
+        self.fileContent = ""
+
+
+class Trie:
+    def __init__(self):
+        self.root = TrieNode()
+        self.endSymbol = "/"
+
+    def addFileContentToPath(self, path, content):
+        currentNode = self.root
+        directories = path.split("/")
+        for i in range(1, len(directories)):
+            directorie = directories[i]
+            currentNode.childDirSet.add(directorie)
+            for char in directorie:
+                if char not in currentNode.children:
+                    currentNode.children[char] = TrieNode()
+                currentNode = currentNode.children[char]
+            currentNode.name = directorie
+        if content:
+            currentNode.childDirSet.add(currentNode.name)
+            currentNode.fileContent += content
+
+    def addPath(self, path):
+        self.addFileContentToPath(path, None)
+
+    def readContentFromPath(self, path):
+        currentNode = self.getLastNode(path)
+        return currentNode.fileContent
+
+    def getCurrentDirContent(self, path):
+        currentNode = self.getLastNode(path)
+        dirList = list(currentNode.childDirSet)
+        dirList.sort()
+        return dirList
+
+    def getLastNode(self, path):
+        currentNode = self.root
+        directories = path.split("/")
+        for i in range(1, len(directories)):
+            currentDir = directories[i]
+            for char in currentDir:
+                currentNode = currentNode.children[char]
+        return currentNode
+
+
+class FileSystem(object):
+
+    def __init__(self):
+        self.trie = Trie()
+
+    def ls(self, path):
+        """
+        :type path: str
+        :rtype: List[str]
+        """
+        return self.trie.getCurrentDirContent(path)
+
+    def mkdir(self, path):
+        """
+        :type path: str
+        :rtype: None
+        """
+        self.trie.addPath(path)
+
+    def addContentToFile(self, filePath, content):
+        """
+        :type filePath: str
+        :type content: str
+        :rtype: None
+        """
+        self.trie.addFileContentToPath(filePath, content)
+
+    def readContentFromFile(self, filePath):
+        """
+        :type filePath: str
+        :rtype: str
+        """
+        return self.trie.readContentFromPath(filePath)
+
+# Your FileSystem object will be instantiated and called as such:
+# obj = FileSystem()
+# param_1 = obj.ls(path)
+# obj.mkdir(path)
+# obj.addContentToFile(filePath,content)
+# param_4 = obj.readContentFromFile(filePath)