Swift中如何实现类似Java TreeMultimap的多值映射功能？

嗨！作为刚入门Swift的新手，碰到这种Java工具类转Swift的问题太正常啦～我来帮你一步步实现类似TreeMultimap的功能！

首先得明确：Swift标准库没有现成的TreeMultimap，但我们可以用基础类型组合+自定义逻辑来实现你要的效果——自动创建键/追加值，同时保证键和值都按自然顺序排序（和Java的Ordering.natural()一致）。

方案1：快速实现基础功能（用Dictionary+扩展） #

如果你只需要核心的“追加值”逻辑，同时保证值列表有序，可以给Dictionary<String, [String]>写个扩展方法：

extension Dictionary where Key == String, Value == [String] {
    // 追加值并自动排序值列表
    mutating func appendAndSort(_ element: String, forKey key: String) {
        var values = self[key] ?? []
        values.append(element)
        values.sort() // 按字符串自然顺序排序，对应Java的Ordering.natural()
        self[key] = values
    }
}

使用起来非常简单，完全符合你的需求：

var multimap: [String: [String]] = [:]

// 第一次添加，自动创建键
multimap.appendAndSort("choosing", forKey: "choose")
// 此时 multimap["choose"] = ["choosing"]

// 第二次添加，自动追加并排序
multimap.appendAndSort("chosen", forKey: "choose")
// 此时 multimap["choose"] = ["chosen", "choosing"]（因为排序后"chosen"在前面）

如果需要按顺序遍历键，只要遍历的时候对键排序就行：

for (key, values) in multimap.sorted(by: { $0.key < $1.key }) {
    print("\(key): \(values)")
}

方案2：自定义TreeMultimap类（完全复刻Java行为） #

如果你需要像Java那样，整个结构始终维护键的自然顺序（而不是遍历临时排序），可以写一个简单的自定义类：

class TreeMultimap {
    // 用字典存储键值对
    private var storage: [String: [String]] = [:]
    // 维护一个始终排序的键数组
    private var sortedKeys: [String] = []
    
    // 对应Java的put方法
    func put(key: String, value: String) {
        // 处理值的排序和追加
        var values = storage[key] ?? []
        values.append(value)
        values.sort()
        storage[key] = values
        
        // 如果是新键，添加后重新排序键数组
        if !sortedKeys.contains(key) {
            sortedKeys.append(key)
            sortedKeys.sort()
        }
    }
    
    // 根据键取值
    func getValues(forKey key: String) -> [String]? {
        return storage[key]
    }
    
    // 获取所有有序的键值对
    func allSortedEntries() -> [(key: String, values: [String])] {
        return sortedKeys.map { ($0, storage[$0]!) }
    }
}

使用示例（对应你Java的代码逻辑）：

// 先定义你的Word结构体
struct Word {
    let type: String
    let word: String
}

// 假设你有third1000数组
let third1000: [Word] = [/* 你的Word实例 */]

// 初始化并填充数据
let multimap = TreeMultimap()
for word in third1000 {
    multimap.put(key: word.type, value: word.word)
}

// 遍历所有有序的键值对
for entry in multimap.allSortedEntries() {
    print("\(entry.key): \(entry.values)")
}

这个类完全复刻了Java TreeMultimap的行为：键始终按自然顺序排列，每个键对应的值列表也是有序的，自动处理键的创建和值的追加。

内容的提问来源于stack exchange，提问作者Nusret Özateş