如何在Scala的List指定位置插入Map中的元素？

给定初始列表和包含插入索引的映射，需要将映射中的键插入到列表对应原索引的位置，最终生成目标列表。以下是可行的实现方案：

步骤1：整理插入任务 #

首先将映射展开为包含所有（原索引，待插入元素）的元组列表，并按原索引升序排序，确保先处理靠前的插入位置，避免后续位置计算出错：

val exampleList = List("A","B","C","D","E","A","L","M","N")
val exampleMap = Map("W1" -> Vector(4,8),
                     "W2" -> Vector(5),
                     "W3" -> Vector(9))

// 展开映射为插入任务列表并排序
val insertions = exampleMap.flatMap { case (key, indices) =>
  indices.map(idx => (idx, key))
}.toList.sortBy(_._1)

步骤2：逐步插入元素 #

使用foldLeft遍历排序后的插入任务，同时维护当前列表和已插入元素的数量（用于调整后续插入位置，因为每次插入会使后续元素的位置偏移1）：

val result = insertions.foldLeft((exampleList, 0)) { case ((currentList, insertedCount), (originalIdx, key)) =>
  // 计算实际插入位置：原索引 + 已插入元素数量
  val insertPos = originalIdx + insertedCount
  // 插入元素并更新状态
  (currentList.insert(insertPos, key), insertedCount + 1)
}._1

// 验证结果
println(result)
// 输出：List(A, B, C, D, W1, E, W2, A, L, M, W1, N, W3)

原理说明 #

• 展开映射是为了将每个键对应的多个索引拆分为独立的插入任务，排序保证了插入操作按原索引从小到大执行，避免位置混乱。
• insertedCount变量跟踪已插入的元素数量，因为Scala的List是不可变的，每次插入都会生成新列表，后续原索引对应的实际位置需要加上已插入的元素数，才能得到正确的插入点。
• List.insert(pos, elem)方法在指定位置插入元素，返回新列表，符合Scala函数式编程的不可变特性。

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