add two more

Author: LEUNGUU

src/search_in_rotated_sorted_array.py

@@ -0,0 +1,27 @@
+from typing import List
+
+
+class Solution:
+    def search(self, nums: List[int], target: int) -> int:
+        left, right = 0, len(nums) - 1
+
+        while left <= right:
+            middle = (left + right) // 2
+
+            if target == nums[middle]:
+                return middle
+
+            # left sorted portion
+            if nums[left] <= nums[middle]:
+                if target < nums[left] or target > nums[middle]:
+                    left = middle + 1
+                else:
+                    right = middle - 1
+            # right sorted portion
+            else:
+                if target > nums[right] or target < nums[middle]:
+                    right = middle - 1
+                else:
+                    left = middle + 1
+
+        return -1

src/time_based_kv_store.py

@@ -0,0 +1,26 @@
+from typing import List, Dict
+
+
+class TimeMap:
+    def __init__(self):
+        self.db: Dict[str, List[str, int]] = {}
+
+    def set(self, key: str, value: str, timestamp: int) -> None:
+        if key not in self.db:
+            self.db[key] = []
+        self.db[key].append([value, timestamp])
+
+    def get(self, key: str, timestamp: int) -> str:
+        res = ""
+        values: List[str, int] = self.db.get(key, [])
+
+        left, right = 0, len(values) - 1
+        while left <= right:
+            middle = (left + right) // 2
+
+            if values[middle][1] <= timestamp:
+                res = values[middle][0]
+                left = middle + 1
+            else:
+                right = middle - 1
+        return res

tests/test_search_in_rotated_sorted_array.py

@@ -0,0 +1,12 @@
+from search_in_rotated_sorted_array import Solution
+
+
+testcase = [{"testcase": [4, 5, 6, 7, 0, 1, 2], "target": 0, "result": 4}]
+
+
+def test_search_in_rotated_sorted_array():
+    for tc in testcase:
+        nums = tc["testcase"]
+        target = tc["target"]
+        ans = Solution().search(nums, target)
+        assert ans == tc["result"]

tests/test_time_based_kv_store.py

@@ -0,0 +1,30 @@
+from time_based_kv_store import TimeMap
+
+testcase = [
+    {
+        "op": ["set", "get", "get", "set", "get", "get"],
+        "value": [
+            ["foo", "bar", 1],
+            ["foo", 1],
+            ["foo", 3],
+            ["foo", "bar2", 4],
+            ["foo", 4],
+            ["foo", 5],
+        ],
+        "result": ["", "bar", "bar", "", "bar2", "bar2"],
+    }
+]
+
+
+def test_time_based_kv_store():
+    tm = TimeMap()
+    for tc in testcase:
+        op_times = len(tc["op"])
+        res = []
+        for index in range(op_times):
+            if tc["op"][index] == "set":
+                tm.set(tc["value"][index][0], tc["value"][index][1], tc["value"][index][2])
+                res.append("")
+            else:
+                res.append(tm.get(tc["value"][index][0], tc["value"][index][1]))
+        assert res == tc["result"]