Scala 代码优化：将输出结果类型从 HashMap 改为 Map

本文将介绍如何将 Scala 代码中输出结果的类型从 HashMap 改为 Map，并提供示例代码和详细解释。

问题描述：

在运行以下代码时，输出结果的第一行是 HashMap(吉林 -> 19, 河南 -> 30, 河北 -> 16)，我们希望将其改为 Map(吉林 -> 19, 河南 -> 30, 河北 -> 16)，即使用 Map 类型来表示输出结果。

示例代码：

object A {
  def main(args: Array[String]): Unit = {
    val data = Map((('吉林', '玉米'), 2),
      (('吉林', '袜子'), 10),
      (('河南', '山药'), 18),
      (('河南', '鞋'), 3),
      (('河南', '衣服'), 9),
      (('河北', '箱包'), 11),
      (('河北', '鸭梨'), 5),
      (('吉林', '木耳'), 7))

    //转换数据格式
    val clickData = data.map { case ((province, product), clickCount) => (province, (product, clickCount)) }

    //按省份分组
    val groupData = clickData.groupBy(_._1)

    //统计每个省份所有商品点击数量的总数
    val countData: Map[String, Int] = groupData.map { case (province, productList) =>
      val clickCount = productList.map(_._2._2).sum
      (province, clickCount)
    }

    //按升序排序
    val sortedData = countData.toList.sortBy(_._2)

    //输出结果
    println(countData)
    println(sortedData)
  }
}

解决方案：

为了将输出结果的类型从 HashMap 改为 Map，需要修改 countData 的类型声明，将其改为 Map[String, Int]。修改后的代码如下：

object A {
  def main(args: Array[String]): Unit = {
    val data = Map((('吉林', '玉米'), 2),
      (('吉林', '袜子'), 10),
      (('河南', '山药'), 18),
      (('河南', '鞋'), 3),
      (('河南', '衣服'), 9),
      (('河北', '箱包'), 11),
      (('河北', '鸭梨'), 5),
      (('吉林', '木耳'), 7))

    //转换数据格式
    val clickData = data.map { case ((province, product), clickCount) => (province, (product, clickCount)) }

    //按省份分组
    val groupData = clickData.groupBy(_._1)

    //统计每个省份所有商品点击数量的总数
    val countData: Map[String, Int] = groupData.map { case (province, productList) =>
      val clickCount = productList.map(_._2._2).sum
      (province, clickCount)
    }

    //按升序排序
    val sortedData = countData.toList.sortBy(_._2)

    //输出结果
    println(countData)
    println(sortedData)
  }
}

解释：

在 Scala 中，HashMap 和 Map 都表示键值对集合。但 HashMap 是 java.util.HashMap 的别名，而 Map 是 Scala 自带的 scala.collection.immutable.Map 的别名。由于 countData 是通过 map 操作从 groupData 生成，而 groupData 的类型是 Map[String, List[(String, (String, Int))]]，所以 countData 的类型也会默认推断为 HashMap[String, Int]。

通过显式地将 countData 的类型声明为 Map[String, Int]，可以强制将输出结果的类型改为 Map。

结论：

通过修改 countData 的类型声明，可以轻松地将输出结果的类型从 HashMap 改为 Map。这只是一个简单的代码优化示例，但它展示了 Scala 类型系统灵活性和强大的功能。