swift 算法简单 62.两个数组的交集II

huanglinxiao

已于 2022-09-29 17:30:06 修改

阅读量685

点赞数

CC 4.0 BY-SA版权

分类专栏： Swift 文章标签：算法

于 2019-06-20 08:49:05 首次发布

本文链接：https://blog.youkuaiyun.com/huanglinxiao/article/details/92960053

Swift 专栏收录该内容

108 篇文章

订阅专栏

博客围绕计算两个数组交集展开，给出示例输入输出，如输入 nums1 = [1,2,2,1]，nums2 = [2,2]，输出 [2,2] 等，并说明输出结果元素出现次数应与原数组一致，不考虑输出顺序，还提及有两种解法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定两个数组，编写一个函数来计算它们的交集。

示例 1:

输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2,2]
示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]
说明：

输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。

解法1：

  func intersect(_ nums1: [Int], _ nums2: [Int]) -> [Int] {
       guard nums1.count > 0 && nums2.count > 0 else {
            return []
        }
        var nums2 = nums2
        
        var array = [Int]()
        for item in 0..<nums1.count {
            
            for (index,value) in nums2.enumerated().reversed() {
                
                if(value == nums1[item]){
                    
                    array.append(nums1[item])
                    nums2.remove(at: index)
                    break

                }
                
            }
       
        }
        
        return array
    }

解法2:

    
    func intersection(_ nums1: [Int], _ nums2: [Int]) -> [Int] {
        guard nums1.count > 0 , nums2.count > 0 else { return [] }
        var array: [Int] = []
        var s1: Set<Int> = Set()
        var s2: Set<Int> = Set()

        nums1.forEach { item in
            s1.insert(item)
        }
        
        nums2.forEach { item in
            s2.insert(item)
        }
        
        if s1.count > s2.count {
            s2.forEach { item in
                if s1.contains(item) {
                    array.append(item)
                }
            }
        } else {
            s1.forEach { item in
                if s2.contains(item) {
                    array.append(item)
                }
            }
        }
        
        
        return array
    }