Scala中的泛型类型推断

时间:2012-12-11 23:31:16

标签: scala generics type-inference

我编写了以下代码,这实际上是scala中的一种愚蠢的合并排序实现:

import scala.collection.immutable.List

object MergeSort {
    def sort[T,E]( comparator: (E,E) => Int ) (l: List[T]): List[T] = {
        def merge[T](first: List[T], second: List[T]): List[T] = (first, second) match {
                case (_, List()) => first
                case (List(), _) => second
                case (f::restFirst, s::restSecond) if comparator(f.asInstanceOf[E],s.asInstanceOf[E]) < 0 => f :: merge(restFirst, second)
                case (f::restFirst, s::restSecond) => s :: merge(first, restSecond)
            }

        l match {
            case List() => return l
            case List(x) => return l
            case _ => {
                val (first, second) = l.splitAt( l.length / 2 )
                merge( sort(comparator)(first), sort(comparator)(second) )
            }
        }
    }
}

这不是以下更优雅的解决方案:

import scala.collection.immutable.List

object MergeSort {
    def sort[T]( comparator: (T,T) => Int ) (l: List[T]): List[T] = {
        def merge[T](first: List[T], second: List[T]): List[T] = (first, second) match {
                case (_, List()) => first
                case (List(), _) => second
                case (f::restFirst, s::restSecond) if comparator(f,s) < 0 => f :: merge(restFirst, second)
                case (f::restFirst, s::restSecond) => s :: merge(first, restSecond)
            }

        l match {
            case List() => return l
            case List(x) => return l
            case _ => {
                val (first, second) = l.splitAt( l.length / 2 )
                merge( sort(comparator)(first), sort(comparator)(second) )
            }
        }
    }
}

哪个不编译,给我以下错误消息:

MergeSort.scala:10: type mismatch;
[error]  found   : f.type (with underlying type T)
[error]  required: T
[error]  case (f::restFirst, s::restSecond) if comparator(f,s) < 0 => f :: merge(restFirst, second)

为什么显式强制转换是必要的,因为底层类型是T?

1 个答案:

答案 0 :(得分:12)

这是我能想到的最烦人的Scala陷阱之一(可能是在与运算符的分号推断相关问题之后)。你是正确答案中的三个字符。

问题是merge上的类型参数。它引入了一个新的T,用于隐藏T上的sort类型参数。因此,编译器不知道comparator可以应用于新T的实例。你可以使用强制转换来管理它,这就是你的第一个版本可以工作的原因,但是否则它会将T视为空白。

只需写下def merge(first: List[T], ...即可。