华为OD机试真题2023_Swift_100_整理扑克牌

// 整理扑克牌
    func HW2023025() {
        // 测试用例
//        let inputStr = "1 1 1 2 3 3 3 4 4 4 5 5 5"
//        let inputStr = "1 3 3 3 2 1 5"
//        let inputStr = "4 4 2 1 2 1 3 3 3 4"
        // 开始代码
        let inputStr = String(readLine()!)
        //  使用高阶函数将字符串转成字典并统计各字符出现次数
        let inputDic = inputStr.replacingOccurrences(of: " ", with: "").reduce(into: [Character: Int]()) { res, key in
            res[key] = Int(res[key] ?? 0) + 1
        }
        // 字典按字符出现次数降序排列，次数一样按字符大小降序排列
        let valueDic = inputDic.sorted { a, b in
            if a.value == b.value { // 张数一样，点数多的排在前面
                return a.key > b.key
            }
            return a.value > b.value  // 张数多的排在前面
        }
        //  将字典转化成数组
        var list: [[Character: Int]] = []
        for (k,v) in valueDic {
            list.append([k: v])
        }
        var res = "" // 记录最后的结果
        var chaiFen: Set<Character> = [] // 需要拆分的点数,可能是多个所以用集合
        for (i,dic) in list.enumerated() {
            var temp = dic
            let carNum = temp.keys.first!
            var carCount = temp.values.first!
            // 当碰到前面是三张，此牌也是三张时，就需要拆分，因为葫芦大于三张
            if i > 0 && list[i-1].values.first == 3 && carCount == 3 {
                chaiFen.insert(carNum) // 将需要拆分的点数存入集合
                carCount = 2 // 将当前点数置为2
                temp.updateValue(2, forKey: carNum) // 将当前点数对应的字典值修改
                list[i] = temp // 修改字典里保存该位置的字典
            }else if carCount == 1 && chaiFen.count != 0 { // 此时是单张，需要注意拆分的牌
                for chai in chaiFen {
                    if chai > carNum { // 如果拆分的牌大于当前单张，则先安排拆分牌
                        res.append("\(String(chai)) ")
                        chaiFen.remove(chai) // 安排完后，将当前拆分牌从集合中移除
                    }
                }
            }
            for _ in 0..<carCount {
                res.append("\(String(carNum)) ")
            }
        }
        if chaiFen.count != 0 { // 如果最后拆分的集合没有安排完，则一次安排完拆分牌
            for chai in chaiFen {
                res.append("\(String(chai)) ")
            }
        }
        print(res.trimmingCharacters(in: .whitespaces))
    }