华为OD机试真题2023_Swift_100_关联端口组合并

// 关联端口组合并
    func HW2023009() {
        var inputArr: [String] = []
        // 测试用例
//        let M = 9
//        inputArr = ["4","2,3,2","1,2","5"]
//        inputArr = ["2,3,1","4,3,2","5"]
//        inputArr = ["10","4,2,1","9","3,6,9,2","6,3,4","8"]
        // 开始代码
        let M = Int(readLine()!)!
        while let line = readLine() {
            inputArr.append(line)
        }
        if M > 10 { // 超过端口组数10的一律打印"[[]]"
            print("[[]]")
        }else {
            var res: [Set<String>] = [] // 端口数组集合
            for i in inputArr {
                let set: Set<String> = Set(i.components(separatedBy: ","))  // 生成要进行合并的端口集合set
                res.append(set) // 将端口集合set直接添加到res
                if set.count != 1 { // 如果端口集合set的元素个数大于1，则需要检查是否需要合并该集合
                    hebing(set, &res, res.count-1)
                }
            }
            
            var resStr: [String] = []
            for i in res {
                // 由于前面集合里面的没有排序，所以这里先排序再拼接
                let temp = Array(i.map { String($0)}).sorted().joined(separator: ",")
                resStr.append("[\(temp)]")
            }
            let temp = resStr.joined(separator: ",")
            print("[\(temp)]") // 打印最终结果
        }
    }
    // set：要进行合并的端口集合 res：端口数组集合 index：需要进行合并的集合下标
    func hebing(_ set: Set<String>, _ res: inout [Set<String>], _ index: Int) {
        var setUnion: Set<String> = [] // 合并后的集合
        for (i,setIndex) in res.enumerated() {
            if i == index { // 因为res已经添加了set，所以当遍历到时直接跳过
                continue
            }
            if set.intersection(setIndex).count >= 2 { // 要进行合并的端口集合set与端口数组集合中遍历出来的的元素setIndex取交集，如果交集个数大于2，则需要进行合并
                setUnion = set.union(setIndex) // 要进行合并的端口集合set与端口数组集合中遍历出来的的元素setIndex取并集setUnion
                let beforeIndex = index > i ? i : index // 求出排在前面的下标
                let afterIndex = index > i ? index : i // 再求排在后面的下标
                res.remove(at: afterIndex) // 先移除后面的集合
                res.remove(at: beforeIndex) // 再移除前面的集合
                res.insert(setUnion, at: beforeIndex) // 再插入合并后的合集到前面的下标位置
                hebing(setUnion, &res, beforeIndex)// 将合并后的合集在进行合并判断，看是否需要再次进行合并
            }
        }
    }