problem solution & test 0001

Author: codewithme-py

README.md

@@ -13,6 +13,6 @@ My clean, typed, and tested solutions to LeetCode problems (Python 3.10+).
 ## Problems
 | # | Title | Difficulty | Solution |
 |---|-------|------------|----------|
-| 1 | [Two Sum](https://leetcode.com/problems/two-sum/) | Easy | [`0001_two_sum.py`](solutions/0001_two_sum.py) |
+| 1 | [Two Sum](https://leetcode.com/problems/two-sum/) | Easy | [`0001_two_sum.py`](solutions/two_sum_0001.py) |
 | 2 | [Add Two Numbers](https://leetcode.com/problems/add-two-numbers/) | Medium | [`0002_add_two_numbers.py`](solutions/0002_add_two_numbers.py) |
 <!-- END_TABLE -->

pyproject.toml

@@ -9,4 +9,5 @@ select = ["E", "W", "F", "I", "UP", "C4"]
 testpaths = ["tests"]
 python_files = ["test_*.py"]
 python_classes = ["Test*"]
-python_functions = ["test_*"]
+python_functions = ["test_*"]
+pythonpath = ["."]

scripts/update_readme.py

@@ -10,7 +10,7 @@
 
 
 def fetch_full_problems_cache():
-    '''Забирает ВСЕ задачи с LeetCode постранично и сохраняет в кэш.'''
+    """Забирает ВСЕ задачи с LeetCode постранично и сохраняет в кэш."""
     print('Get all LeetCode problems...')
     all_questions = []
     skip = 0
@@ -62,20 +62,22 @@ def fetch_full_problems_cache():
 
 
 def load_problems_cache():
-    '''Загружает кэш или создаёт новый (полный).'''
+    """Загружает кэш или создаёт новый (полный)."""
     if Path(CACHE_FILE).exists():
         with open(CACHE_FILE, 'r') as f:
             return json.load(f)
     return fetch_full_problems_cache()
 
-
 def extract_problem_number(file_name: str) -> int | None:
-    '''Извлекает номер задачи из имени файла.'''
-    if not file_name.endswith('.py'):
+    """Извлекает номер задачи из имени файла."""
+    if not file_name.endswith('.py') or not file_name[:-3].replace('_', '').isdigit():
+        return None
+    parts = file_name[:-3].split('_')
+    if not parts:
         return None
     try:
-        return int(file_name.split('_')[0])
-    except (ValueError, IndexError):
+        return int(parts[-1])
+    except ValueError:
         return None
 
 

solutions/0001_two_sum.py

@@ -0,0 +1,17 @@
+from typing import List
+
+class Solution:
+    def twoSum(self, nums: List[int], target: int) -> List[int]:
+        """Возвращает индексы двух чисел из nums, сумма которых равна target.
+        Работает за O(n): проходит по списку один раз, запоминая числа
+        и их позиции в словаре. Для каждого элемента проверяет,
+        было ли уже число (target - текущее). Если да — возвращает пару индексов.
+        Гарантируется, что решение существует и уникально.
+        """
+        seen = {}
+        for index, num in enumerate(nums):
+            complement = target - num
+            if complement in seen:
+                return [seen[complement], index]
+            seen[num] = index
+        raise ValueError('Нет решения')

solutions/0002_add_two_numbers.py

@@ -0,0 +1,7 @@
+from solutions.two_sum_0001 import Solution
+
+def test_two_sum():
+    sol = Solution()
+    assert sol.twoSum([2, 7, 11, 15], 9) == [0, 1]
+    assert sol.twoSum([3, 2, 4], 6) == [1, 2]
+    assert sol.twoSum([3, 3], 6) == [0, 1]