1. 程式人生 > >LeetCode刷題記錄——第349題(兩個陣列的交集)

LeetCode刷題記錄——第349題(兩個陣列的交集)

題目描述

給定兩個陣列,編寫一個函式來計算它們的交集。

示例 1:

輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2]

示例 2:

輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [9,4]

思路分析

  • 根據題目可知,求交集,返回的是一個列表,自然而然的想到用一個新列表,使用append方法將符合要求的元素append進去。
  • 滿足條件的元素為:同時在nums1和nums2中,並且在result中未出現。所以可以利用for迴圈遍歷nums1中的所有元素,然後用if將滿足條件的執行append方法。bingo!

程式碼示例

class Solution(object):
    def intersection(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        result = []
        for i in nums1:
            if i in nums2 and i not in result:
                result.
append(i) return result