我正在尝试使用 TreeMap 解决编程问题,这导致我发布了这个问题。
我定义了一个以 (Int,Int) 为键,以 String 为值的 treeMap,根据元组的第一个元素定义了 treeMap 排序。并且我插入了两个不同Key的元素,但最终的treeMap只包含一个元素。这是treeMap的定义行为吗
Scala 代码:(版本:2.12.3)
val a = scala.collection.mutable.TreeMap[(Int, Int), String]()(Ordering.by(x => x._1))
a.put((9, 21), "value-1")
a.put((9, 10), "value-2")
println(a.size) // 1
Run Code Online (Sandbox Code Playgroud)
我已经在 java 中尝试了相同的实现,但它报告 Map Size 为 2,这是我的 java 代码:
如果我遗漏了一些东西,有人可以建议吗
import java.util.Comparator;
import java.util.TreeMap;
public class JavaTreeMapTest {
public static void main(String[] args) {
class Tuple {
Integer a;
Integer b;
public Tuple(Integer a, Integer b) {
this.a = a;
this.b = b;
}
}
Comparator<Tuple> testComparator = new Comparator<Tuple>() {
@Override
public int compare(Tuple arg0, Tuple arg1) {
if (arg0.a > arg1.a) {
return arg0.a;
} else
return arg1.a;
}
};
TreeMap<Tuple, String> tm = new TreeMap<Tuple, String>(testComparator);
tm.put(new Tuple(100, 100), "value-1");
tm.put(new Tuple(100, 101), "value-2");
System.out.println(tm.size()); //2
}
}
Run Code Online (Sandbox Code Playgroud)
问题出在您的订购上。如果像这样使用它:Ordering.by(x => x._1),TreeMap 将只考虑第一个元素来计算相等性。因此,两者(9, 21)和(9, 21)将被视为相等。
一种解决方案是先按第一个元素排序,然后按第二个元素排序。您可以通过返回一个元组做到这一点:
Ordering.by(x => (x._1, x._2))。
但由于您已经在使用元组,您可以将其简化为:
Ordering.by(x => x)
Run Code Online (Sandbox Code Playgroud)
或者
Ordering.by(identity)
Run Code Online (Sandbox Code Playgroud)
或者
您可以省略它,因为按身份排序是默认设置。
总结:
val a = scala.collection.mutable.TreeMap[(Int, Int), String]()
a.put((9, 21), "value-1")
a.put((9, 10), "value-2")
println(a.size) // 2
Run Code Online (Sandbox Code Playgroud)