欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

lintcode:两个数组的交

程序员文章站 2022-03-24 17:44:44
...

lintcode:两个数组的交

class Solution:
    # @param {int[]} nums1 an integer array
    # @param {int[]} nums2 an integer array
    # @return {int[]} an integer array
    def intersection(self, nums1, nums2):
        # Write your code here

        # 方法一:
        A, B = len(nums1), len(nums2)
        nums1.sort()
        nums2.sort()
        a = []
        i, j = 0, 0
        while i < A and j < B:
            if nums1[i] < nums2[j]:
                i += 1
            elif nums1[i] == nums2[j]:
                a.append(nums1[i])
                i += 1
                j += 1
            else:
                j += 1
        a = list(set(a))
        return a
class Solution:
    # @param {int[]} nums1 an integer array
    # @param {int[]} nums2 an integer array
    # @return {int[]} an integer array
    def intersection(self, nums1, nums2):
        # Write your code here

        # 方法二:set
        nums1, nums2 = set(nums1), set(nums2)
        a = nums1 & nums2
        return list(a)