如何用Swift高阶函数实现给定和的数对查找？

这里提供两种基于高阶函数的实现方案，核心逻辑和你原来的迭代方式一致，只是用函数式写法更简洁：

方案一：生成所有可能数对后筛选 #

let array = [1,2,3,4,5]
let givenSum = 9

// 生成所有i<j的数对，再筛选和为指定值的结果
let targetPairs = array.enumerated()
    .flatMap { index, currentNum in
        // 对当前元素之后的所有元素，生成(currentNum, 后续元素)的元组
        array[(index + 1)...].map { (currentNum, $0) }
    }
    .filter { $0.0 + $0.1 == givenSum }

// 如果需要转成数组格式而非元组
let targetArrays = targetPairs.map { [$0.0, $0.1] }

print(targetPairs)    // 输出 [(4, 5)]
print(targetArrays)   // 输出 [[4, 5]]

步骤解释： #

• enumerated()：获取每个元素的索引和对应值，确保我们只取当前元素之后的元素配对，避免重复（比如不会出现[5,4]这种反向对）
• flatMap：将每个元素对应的后续元素对展开成一个一维数组，替代外层循环
• map：生成当前元素和后续元素的配对，替代内层循环
• filter：筛选出和为给定值的数对

方案二：更紧凑的写法（合并map和filter） #

如果想减少中间步骤，可以把筛选逻辑整合到flatMap里：

let array = [1,2,3,4,5]
let givenSum = 9

let targetPairs = array.enumerated().flatMap { index, currentNum in
    array[(index + 1)...].compactMap { nextNum in
        currentNum + nextNum == givenSum ? (currentNum, nextNum) : nil
    }
}

print(targetPairs) // 输出 [(4, 5)]

这里用compactMap替代map+filter，直接在生成配对时过滤掉不符合条件的结果，不需要额外的filter步骤。

内容的提问来源于stack exchange，提问作者Ajay K