做Leetcode题时, 经常需要对数组进行排序,这时候可以使用Arrays.sort(arr, new Comparator()){}
例如: 对 new int[] intervals 进行排序,排序后的结果会更新到intervals中
Arrays.sort(intervals, new Comparator<int[]>() {
@Override
public int compare(int[] o1, int[] o2) {
if (o1[0] > o2[0]) {
return 1;
} else if (o1[0] == o2[0]) {
if (o1[1] < o2[1]) {
return -1;
} else {
return 1;
}
} else {
return -1;
}
}
});
Comparator
是一个Java中一个高度抽象的接口,这个接口可以实现排序功能,还有其他比较实用的功能,见另一篇博客.
对于实现Comparator
接口的compare()
方法来说返回值为正则传入的第一个对象大于第二个对象,返回值为0表示传入的两个对象一样大;返回值为负表示传入的第一个对象小于第二个对象。
在实现Comparator
接口时,需要注意实现compare方法满足的三个性质:
自反性 两个相等的元素相比时,compare必须返回0
反对称性 compare(o1,o2)=1=>compare(o2,o1)=-1
传递性 compare(o1,o2)=1, compare(o2,o3)=1 => compare(o1,o3)=1
否则会报这样的错误:
Comparison method violates its general contract!
java.lang.IllegalArgumentException: Comparison method violates its general contract!
at line 781, java.base/java.util.TimSort.mergeLo
at line 518, java.base/java.util.TimSort.mergeAt
at line 448, java.base/java.util.TimSort.mergeCollapse
at line 245, java.base/java.util.TimSort.sort
at line 1442, java.base/java.util.Arrays.sort
at line 4, Solution.merge
at line 54, __DriverSolution__.__helper__
at line 84, __Driver__.main
不满足上述三个性质的compare方法,运行时就会报Comparison method violates its general contract!错误.
比如举例代码就违反了自反性,当两个元素相等的时候,没有返回0,改成如下代码才行:
public int compare(int[] o1, int[] o2) {
if (o1[0] > o2[0]) {
return 1;
} else if (o1[0] == o2[0]) {
return o1[1] - o2[1];
} else {
return -1;
}
}