ZhongKuo0228 / study

0 stars 0 forks source link

2215. Find the Difference of Two Arrays #78

Open fockspaces opened 10 months ago

fockspaces commented 10 months ago
  1. 建立兩個 set,並額外建立一個 set 作為 舊有的 set_1
  2. 迭代兩 list,如果有在另一方找到,則 delete 自己的 set
  3. 最後留下的 set_1, set_2 即為獨有的 elements
class Solution:
    def findDifference(self, nums1: List[int], nums2: List[int]) -> List[List[int]]:
        set_1, set_2 = set(nums1), set(nums2)
        old_set_1 = set(set_1)
        for num in nums1:
            if num in set_2 and num in set_1:
                set_1.remove(num)
        for num in nums2:
            if num in old_set_1 and num in set_2:
                set_2.remove(num)
        return [list(set_1), list(set_2)]
fockspaces commented 10 months ago

GPT improve 太騷操作了... 學不來

class Solution:
    def findDifference(self, nums1: List[int], nums2: List[int]) -> List[List[int]]:
        set_1, set_2 = set(nums1), set(nums2)
        return [set_1 - set_2, set_2 - set_1]