在 Swift 中通过计数从多维数组中找到最大的数组

Question

我有一个多维数组。它本身可以有多个数组。有时可能是 5 或 6。现在我想通过使用数组大小​​从我的多维数组中获取最大的数组。我不知道如何实现这一点。因此我在这里发帖。提前致谢。

例如：

[[“a”、“b”、“c”]、[“d”、“e”、“f”、“g”、“h”、“i”、“j”]]]

Answer 1

如果您正在寻找给定的最长子数组 数组，那么您可以简单地使用max(by:) 进行比较 使用数组计数：

let a = [["a", "b", "c"], ["d", "e", "f", "g", "h", "i", "j"], ["k"]]

let longestSubArray = a.max(by: { $0.count < $1.count })!

print(longestSubArray)
// ["d", "e", "f", "g", "h", "i", "j"]

这里我假设a不为空，否则max(by:) 将返回nil。如果发生这种情况，请使用可选绑定：

if let longestSubArray = a.max(by: { $0.count < $1.count }) {
    print(longestSubArray)
} else {
    print("a is empty")
}

备注： Array 是 RandomAccessCollection，因此得到 它的count 是O(1) 操作。

如果你需要最长的元素和它在包含中的索引 数组然后您可以将上述应用到a.enumerated():

if let (idx, longest) = a.enumerated().max(by: { $0.element.count < $1.element.count }) {
    print("longest subarray", longest)
    print("at index", idx)
}

如果有超过一个最大长度的子数组，那么上面的 解决方案将返回其中之一。 @dfri's answer 显示如何获取 all 具有最大长度的子数组。

Answer 2

也许是最明显（不是最短）的方式：

我只是循环遍历数组并检查里面数组的大小

let anArray = [["Hello", "This", "is"], ["Hello", "This", "is", "a", "5"], ["Hello", "This", "is", "a", "6", "!"]])

var max = 0

for array in anArray
{
    if array.count > max
    {
        max = array.count
    }
}

print(max)

您可以根据自己的目的使用 AnyObject 数组或其他东西来代替字符串

编辑

根据您的 cmets，只需进行以下更新：

我做了这个函数：

func getLongestArray(anArr:[[String]]) -> [String]
{
    var maxArr = [String]()

    for array in anArr
    {
        if(array.count > maxArr.count)
        {
            maxArr = array
        }
    }

    return maxArr
}

然后在适当的情况下进行此调用

let anArray = [["Hello", "This", "is"], ["Hello", "This", "is", "a", "5"], ["Hello", "This", "is", "a", "6", "!"]]

print(getLongestArray(anArr: anArray))

这应该返回最长的数组，尽管假设您至少有一个大小 > 0 的数组

Answer 3

@MartinR:s answer 中使用的 max(by:) 方法在这里是适合目的的解决方案，但是（正如我通常写的 :)，为了获得替代方案的乐趣，您还可以“手动”折叠数组上的索引以找到最长子数组对应的索引（最大.count）：

let arr = [["a", "b", "c"], ["d", "e", "f", "g", "h", "i", "j"]]

if arr.count > 0 {
    let longestSubArray = arr[
        arr.indices.reduce(0) { arr[$1].count > arr[$0].count ? $1 : $0 } // (*)
    ]
    print(longestSubArray) // ["d", "e", "f", "g", "h", "i", "j"]
}

/* (*) Alternatively:
arr.indices.lazy.map { arr[$0].count }.max() ?? 0 */

如果您想找到计数最多的子数组子集（可能不是单个子数组），您可以使用类似的方法并结合filter：

// two longest subarrays with same count
let arr = [["a", "b", "c"],
           ["d", "e", "f", "g", "h", "i", "j"],
           ["k", "l", "m", "n", "o", "p", "q"]]

let maxSubArrayCount = arr.lazy.map { $0.count }.max() // thanks @muescha (+)
let longestSubArrays = arr.filter { $0.count == maxSubArrayCount }
print(longestSubArrays) /* [["d", "e", "f", "g", "h", "i", "j"], 
                            ["k", "l", "m", "n", "o", "p", "q"]] */

/* (+) Alternatively:
let maxSubArrayCount = arr.reduce(0) { $1.count > $0 ? $1.count : $0 } */