feat: [LeetCode #2215] Find The Difference Of Two Arrays (#7)

Тип: HashMap / Set
Сложность: easy
Временная сложность: O(N + M)
Пространственная сложность: O(N + M)

-

Ссылка: https://leetcode.com/problems/find-the-difference-of-two-arrays/

Author: onixlas

src/hashmap_set/__init__.py

@@ -0,0 +1,12 @@
+from typing import List
+
+
+class Solution:
+    def findDifference(self, nums1: List[int], nums2: List[int]) -> List[List[int]]:
+        set1 = set(nums1)
+        set2 = set(nums2)
+
+        ans1 = set(num for num in nums1 if num not in set2)
+        ans2 = set(num for num in nums2 if num not in set1)
+
+        return [list(ans1), list(ans2)]

src/hashmap_set/find_the_difference_of_two_arrays/__init__.py

@@ -0,0 +1,11 @@
+import pytest
+from src.hashmap_set.find_the_difference_of_two_arrays.solution import Solution
+
+
+@pytest.mark.parametrize(
+    "nums1, nums2, expected",
+    [([1, 2, 3], [2, 4, 6], [[1, 3], [4, 6]]), ([1, 2, 3, 3], [1, 1, 2, 2], [[3], []])],
+)
+def test_find_the_difference_of_two_arrays(nums1, nums2, expected):
+    solution = Solution()
+    assert solution.findDifference(nums1, nums2) == expected