scala-collections - Scala：如何对不同长度列表中具有相同索引的所有元素求和？

Question

我有一个整数列表列表。需要获取列中所有元素的总和列表，从右到左，如果列中的值不存在（一行比其他行短），则设为 0：

            ( 1,  7),
    ( 3,  4,  6,  0),
                  (),
( 9,  1,  6,  7,  2),
                 ...
---------------------
( 9,  4, 10, 14,  9)

我是用 Java 写的（比如说，一种命令式语言）：

    int[][] s = {
                 { 1, 7 },
           { 3, 4, 6, 0 },
                       {},
        { 9, 1, 6, 7, 2 }
    };
    
    int maxLen = 0;
    for (int[] x : s) {
      maxLen = Integer.max(maxLen, x.length);
    }
    
    int[] sum = new int[maxLen];
    for (int j = maxLen; j-- > 0; ) {
      for (int[] i : s) {
        int idxFromEnd = j - maxLen + i.length;
        sum[j] += idxFromEnd >= 0 ? i[idxFromEnd] : 0;
      }
    }

然后翻译成Scala：

  val s = List(
             List( 1, 7 ),
       List( 3, 4, 6, 0 ),
                   List(),
    List( 9, 1, 6, 7, 2 ),
  )

  val maxLen = s.view.map(_.length).max
  val sum =
    for (i <- 0 until maxLen)
    yield s.view.map(row => {
      val idxFromEnd = i - maxLen + row.length
      row.lift(idxFromEnd).getOrElse(0)
    }).sum

请你们建议一种用 Scala 编写它的惯用方式。尾递归也是可以接受的。

标签： scala-collections

Answer 1

您可以使用groupBy和flatMap：

val s = List(
             List( 1, 7 ),
             List( 3, 4, 6, 0 ),
             List(),
             List( 9, 1, 6, 7, 2 ),
            )

s.flatMap(_.zipWithIndex).groupBy{case (num, idx) => idx}
 .map{case (k, v) => v.map(_._1).sum}.toList