349. 两个数组的交集

class Solution(object):
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        # 保证s1短
        if len(nums1) > len(nums2):
            nums1, nums2 = nums2, nums1
        # 返回值
        res = [int] * len(nums1)
        # res下标
        index = 0
        # 遍历s1
        i = 0
        while i < len(nums1):
            # 在s2中，不在res中，则添加到res中，下标加1，指针后移
            if nums1[i] in nums2 and nums1[i] not in res:
                res[index] = nums1[i]
                index += 1
                i += 1
            # 指针后移
            else:
                i += 1
        return res[0:index]


if __name__ == '__main__':
    solution = Solution()
    print(solution.intersection(nums1=[4, 9, 5], nums2=[9, 4, 9, 8, 4]))
    print(solution.intersection(nums1=[1, 2, 2, 1], nums2=[2, 2]))