给定以下列表:“A”、“B”、“C”、“D”、“E”、“F”、“G”
我需要一个进行以下排序的比较器:
- 指定某个元素(例如
“D”
) - 从元素开始
- 后跟原始列表中的所有后续元素(按原始顺序)
- 按照原始顺序跟随原始列表的所有前面的元素
结果将是:“D”、“E”、“F”、“G”、“A”、“B”、“C”
请注意,我知道我可以做类似于以下的事情:
List<String> following = myList.subList(myList.indexOf("D") + 1, myList.size());
List<String> preceding = myList.subList(0, myList.indexOf("D"));
List<String> newList = Stream.of(Collections.singletonList("D"), following, preceding)
.flatMap(List::stream)
.collect(Collectors.toList());
在这个问题中,我明确指的是 Comparator
实现。
很明显它必须将列表和元素作为参数,我只是不清楚比较算法本身:
private static class MyComparator<T> implements Comparator<T> {
private final List<T> list;
private final T element;
private MyComparator(List<T> list, T element) {
this.list = list;
this.element = element;
}
@Override
public int compare(T o1, T o2) {
// Not clear
}
}
最佳答案
我想这就是你想要的:
class ImposedOrder<T> implements Comparator<T> {
private final List<T> list;
private final int startIndex;
ImposedOrder(List<T> list, T startElement) {
this.list = new ArrayList<>(list);
this.startIndex = list.indexOf(startElement);
if (startIndex < 0) {
throw new IllegalArgumentException();
}
}
@Override
public int compare(T t1, T t2) {
int t1Index = list.indexOf(t1);
int t2Index = list.indexOf(t2);
return Integer.compare(adjust(t1Index), adjust(t2Index));
}
private int adjust(int rawIndex) {
if (rawIndex >= startIndex) {
return rawIndex;
}
return rawIndex + list.size();
}
}
一些额外的验证可能是为了避免强加的重复顺序列表。
使用 indexOf
的线性搜索不会给您带来很好的性能,但对于小订单列表来说可能就足够了。否则,您可以将元素映射到比较器构造函数中调整后的索引,而不是保存强加顺序列表的副本。
像这样:
class ImposedOrder<T> implements Comparator<T> {
private final Map<T, Integer> map;
private final int startIndex;
ImposedOrder(List<T> list, T startElement) {
this.startIndex = list.indexOf(startElement);
if (startIndex < 0) {
throw new IllegalArgumentException();
}
this.map = IntStream.range(0, list.size())
.boxed()
.collect(Collectors.toMap(
list::get,
i -> adjust(startIndex, list.size(), i)
));
}
@Override
public int compare(T t1, T t2) {
Integer t1Index = map.get(t1);
Integer t2Index = map.get(t2);
return t1Index.compareTo(t2Index);
}
private static int adjust(int startIndex, int size, int rawIndex) {
if (rawIndex >= startIndex) {
return rawIndex;
}
return rawIndex + size;
}
}
关于java - 自定义比较器,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44300364/