Kotlin中如何将Int列表压缩为支持逆向还原的Int范围列表？

嘿，先给你点个赞，你已经摸到了核心思路——把单个整数转成单元素范围，再合并相邻的连续范围！不过你的代码还有几个可以优化的点，而且我们得确保完全能逆向还原（包括保留重复元素），我来一步步帮你完善：

现有代码的小隐患 #

你的逻辑方向是对的，但有个潜在问题：如果输入的ints是空列表，ints[0]会直接抛出IndexOutOfBoundsException。另外，用ints[0]..(ints[0]-1)这种空范围启动fold，虽然最终能得到正确结果，但可读性稍差，我们可以换一种更直观的初始值处理方式。

优化后的压缩代码 #

我们先判断列表是否为空，再从第一个元素的范围开始fold，逻辑更清晰，也避免了空列表崩溃的问题：

fun compressIntsToRanges(ints: List<Int>): List<IntRange> {
    if (ints.isEmpty()) return emptyList()
    
    return ints.map { it..it }
        .fold(mutableListOf(ints.first()..ints.first())) { acc, next ->
            val prev = acc.last()
            when {
                // 当前范围是上一个范围的连续延伸，合并范围
                prev.last + 1 == next.first -> {
                    acc[acc.lastIndex] = prev.first..next.last
                }
                // 处理重复元素：直接添加相同范围，确保还原时能恢复重复项
                prev == next -> {
                    acc.add(next)
                }
                // 非连续也非重复，添加新范围
                else -> {
                    acc.add(next)
                }
            }
            acc
        }.toList()
}

// 测试你的示例输入
val ints = listOf(8, 9, 45, 48, 49, 60, 61, 61, 62, 63, 3, 4, 5, 4, 5, 6)
val compressedRanges = compressIntsToRanges(ints)
println(compressedRanges)
// 输出：[8..9, 45..45, 48..49, 60..61, 61..63, 3..5, 4..6]

这里特意保留了重复范围的添加逻辑，只有这样，逆向还原时才能恢复原列表里的重复元素（比如原列表的两个61，压缩后对应60..61和61..63的重叠部分，展开后就能得到两个61）。

逆向还原函数 #

既然要求不能丢失信息，我们必须实现还原函数，把压缩后的范围列表转回原整数列表：

fun decompressRangesToInts(ranges: List<IntRange>): List<Int> {
    return ranges.flatMap { range ->
        // 把每个范围展开成整数序列
        range.toList()
    }
}

// 测试还原是否正确
val decompressedInts = decompressRangesToInts(compressedRanges)
println(decompressedInts == ints) // 输出：true

这个函数逻辑很简单：把每个IntRange展开成整数列表再拼接，完全能恢复原列表的所有元素，包括重复项。

边界情况处理 #

优化后的代码已经覆盖了空列表和单元素列表的情况：

println(compressIntsToRanges(emptyList())) // 输出：[]
println(compressIntsToRanges(listOf(5))) // 输出：[5..5]
println(decompressRangesToInts(listOf(5..5)) == listOf(5)) // 输出：true

内容的提问来源于stack exchange，提问作者Michał Šrajer