daily

Author: TurtleSmoke

problems/problem_0796/solution_1.py

@@ -0,0 +1,32 @@
+import pytest
+
+
+class Solution:
+    def rotateString(self, s: str, goal: str) -> bool:
+        for i in range(len(s)):
+            if s[i:] + s[:i] == goal:
+                return True
+
+        return False
+
+
+tests = [
+    (
+        ("abcde", "cdeab"),
+        True,
+    ),
+    (
+        ("abcde", "abced"),
+        False,
+    ),
+]
+
+
+@pytest.mark.timeout(2)
+@pytest.mark.parametrize(
+    "inputs, expected",
+    tests,
+)
+def test_validator(inputs, expected):
+    output = Solution().rotateString(*inputs)
+    assert output == expected

problems/problem_0796/solution_2.py

@@ -0,0 +1,28 @@
+import pytest
+
+
+class Solution:
+    def rotateString(self, s: str, goal: str) -> bool:
+        return len(s) == len(goal) and goal in s + s
+
+
+tests = [
+    (
+        ("abcde", "cdeab"),
+        True,
+    ),
+    (
+        ("abcde", "abced"),
+        False,
+    ),
+]
+
+
+@pytest.mark.timeout(2)
+@pytest.mark.parametrize(
+    "inputs, expected",
+    tests,
+)
+def test_validator(inputs, expected):
+    output = Solution().rotateString(*inputs)
+    assert output == expected

problems/problem_1233/solution_1.py

@@ -0,0 +1,55 @@
+from collections import defaultdict
+from typing import List
+
+import pytest
+
+
+class Solution:
+    class Trie:
+        def __init__(self):
+            self.children = defaultdict(Solution.Trie)
+            self.word = ""
+
+    def removeSubfolders(self, folder: List[str]) -> List[str]:
+        root = Solution.Trie()
+        for path in folder:
+            node = root
+            sub_folders = path.split("/")
+            for sub_folder in sub_folders[1:]:
+                node = node.children[sub_folder]
+            node.word = path
+
+        def dfs(node):
+            if node.word:
+                yield node.word
+            else:
+                for child in node.children:
+                    yield from dfs(node.children[child])
+
+        return list(dfs(root))
+
+
+tests = [
+    (
+        (["/a", "/a/b", "/c/d", "/c/d/e", "/c/f"],),
+        ["/a", "/c/d", "/c/f"],
+    ),
+    (
+        (["/a", "/a/b/c", "/a/b/d"],),
+        ["/a"],
+    ),
+    (
+        (["/a/b/c", "/a/b/ca", "/a/b/d"],),
+        ["/a/b/c", "/a/b/ca", "/a/b/d"],
+    ),
+]
+
+
+@pytest.mark.timeout(2)
+@pytest.mark.parametrize(
+    "inputs, expected",
+    tests,
+)
+def test_validator(inputs, expected):
+    output = Solution().removeSubfolders(*inputs)
+    assert output == expected

problems/problem_3163/solution_1.py

@@ -0,0 +1,37 @@
+import pytest
+
+
+class Solution:
+    def compressedString(self, word: str) -> str:
+        res = ""
+        i = 0
+        while i < len(word):
+            j = i
+            while j < len(word) and (j - i) < 9 and word[i] == word[j]:
+                j += 1
+            res += str(j - i) + word[i]
+            i = j
+        return res
+
+
+tests = [
+    (
+        ("abcde",),
+        "1a1b1c1d1e",
+    ),
+    (
+        ("aaaaaaaaaaaaaabb",),
+        "9a5a2b",
+    ),
+    (
+        ("abcde",),
+        "1a1b1c1d1e",
+    ),
+]
+
+
+@pytest.mark.timeout(2)
+@pytest.mark.parametrize("inputs, expected", tests)
+def test_validator(inputs, expected):
+    output = Solution().compressedString(*inputs)
+    assert output == expected