将数组中的重复元素配对

Question

我正在尝试将数组的重复元素配对并计数。

当给定数组是：[10, 20, 20, 10, 10, 30, 50, 10, 20] 时，我期望 numberOfPairs 为 3。因为有 2 对 10 和 1 对 20。

我的“if 条件”正在检查当前元素的索引是否是第一个索引。如果不是最后一个索引，则表示当前元素存在重复。所以我给 numberOfPairs 加 1。

对于输入 [10, 20, 20, 10, 10, 30, 50, 10, 20]，我的 numberOfPairs 是 2 但应该是 3。 对于输入 [1 1 3 1 2 1 3 3 3 3]，myNumberOfPairs 根本不打印？但它应该是 4。

我在这里错过了什么？

func sockMerchant(n: Int, ar: [Int]) -> Int {
    // Write your code here
    var array = ar
    var numberOfPairs = 0
    
    for i in 0..<array.count {
       var element = array[i]
       let indexOfLastElement = array.lastIndex(of: element)
       let indexOfFirstElement = array.firstIndex(of: element)
       print("indexOfLastElement is \(indexOfLastElement)")
       print("indexOfFirstElement is \(indexOfFirstElement)")
       
       if indexOfFirstElement != indexOfLastElement {
           numberOfPairs += 1
           array.remove(at: indexOfFirstElement!)
           array.remove(at: indexOfLastElement!)
           continue
       }
       return numberOfPairs
    }
     return numberOfPairs
 }

Answer 1

您在访问remove(at:) 的同时调用remove(at:) 来改变您的array，这就是为什么您会出现这些奇怪的副作用。

我假设您正在尝试解决 Leetcode 任务（或类似任务），因此我不会预先提供解决方案。我对您的建议是考虑一种算法，该算法在您阅读同一列表的这些内容时不涉及更改列表的内容。

Answer 2

所以，感谢@Vym 和@Martin R，我已经解决了以下问题。

func sockMerchant(n: Int, ar: [Int]) -> Int {
// Write your code here
var array = ar
var numberOfPairs = 0
var newArray = [Int]()
var done = false

for i in 0..<array.count {
   let element = array[i]
   let indexOfLastElement = array.lastIndex(of: element)
   let indexOfFirstElement = array.firstIndex(of: element)
   
   if indexOfFirstElement != indexOfLastElement {
       newArray.append(element)
       numberOfPairs = newArray.count/2
       done = true
   }
   
}
if done == true {
    return numberOfPairs
}
 return numberOfPairs

}

Answer 3

我同意@MartinR 的观点，在这种情况下，您应该放置断点并逐行遍历您的代码，很高兴您自己发现了错误。

但在性能方面，lastIndex 和 firstIndex 是非常繁重的操作，因为它们可能会考虑所有项目而一无所获，这使得 Big O notation 的算法在 O(log n) 左右。在这种情况下，字典被广泛使用（如果您对空间没有太多限制）。

您可以将 value 用作 key 并将其作为 value 用于字典并计算所有项目，然后像这样求和：

func sockMerchant(ar: [Int]) -> Int {
    ar.reduce(into: [Int:Int]()) { map, value in
        map[value, default: 0] += 1
    }.reduce(0) { sum, count in
        sum + count.value / 2
    }
 }