这是 (1) 一种非常有效的通用方法,以及 (2) 一种简单有效的方法。
(1) 在第一个排序中,我计算并缓存数组中每个值一次的数字 5 的出现次数。然后我使用通用排序方法根据缓存的值进行排序。这比自定义比较器要好,因为列表中的每个元素只计算一次出现的次数。
(2)在第二种排序中,我只是使用了一个自定义比较器,实现起来更简单。
排序 (3) 和 (4) 分别使用字符串而不是整数执行与 (1) 和 (2) 相同的操作。
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.ListIterator;
public class DigitSort {
public static void main(String[] args) {
List<Integer> initInt = Arrays.asList(555, 55, 5, 101, 202);
final int sortDigit = 5;
// sort #1
List<Integer> ints = new ArrayList<>(initInt);
sort(ints, new ToComparable<Integer, DigitComparable>() {
@Override
public DigitComparable toComparable(Integer number) {
return number == null ? null : new DigitComparable(number, digitCount(sortDigit, number));
}
});
System.out.println("Results of int generic sort:");
for (Integer number : ints)
System.out.println(number);
// sort #2
ints = new ArrayList<>(initInt);
Collections.sort(ints, new Comparator<Integer>() {
@Override
public int compare(Integer number1, Integer number2) {
if (number1 == null && number2 == null)
return 0;
if (number1 == null)
return -1;
if (number2 == null)
return 1;
int digitCount1 = digitCount(sortDigit, number1);
int digitCount2 = digitCount(sortDigit, number2);
return
digitCount1 > 0 && digitCount2 > 0 ? digitCount1 - digitCount2 :
digitCount2 > 0 ? -1 :
digitCount1 > 0 ? 1 :
number1 - number2;
}
});
System.out.println("\nResults of int comparator sort:");
for (Integer number : ints)
System.out.println(number);
List<String> initString = new ArrayList<>(initInt.size());
for (Integer number : initInt)
initString.add(number == null ? null : number.toString());
// sort #3
List<String> strings = new ArrayList<>(initString);
sort(strings, new ToComparable<String, DigitComparable>() {
@Override
public DigitComparable toComparable(String number) {
if (number == null)
return null;
int asInt = Integer.parseInt(number);
return new DigitComparable(asInt, digitCount(sortDigit, asInt));
}
});
System.out.println("\nResults of string generic sort:");
for (String number : strings)
System.out.println(number);
// sort #4
strings = new ArrayList<>(initString);
Collections.sort(strings, new Comparator<String>() {
@Override
public int compare(String number1, String number2) {
if (number1 == null && number2 == null)
return 0;
if (number1 == null)
return -1;
if (number2 == null)
return 1;
int digitCount1 = digitCount(sortDigit, number1);
int digitCount2 = digitCount(sortDigit, number2);
return
digitCount1 > 0 && digitCount2 > 0 ? digitCount1 - digitCount2 :
digitCount2 > 0 ? -1 :
digitCount1 > 0 ? 1 :
Integer.parseInt(number1) - Integer.parseInt(number2);
}
});
System.out.println("\nResults of string comparator sort:");
for (String number : strings)
System.out.println(number);
}
private static class DigitComparable implements Comparable<DigitComparable> {
private final int number, digitCount;
DigitComparable(int number, int digitCount) {
this.number = number;
this.digitCount = digitCount;
}
@Override
public int compareTo(DigitComparable other) {
return
digitCount > 0 && other.digitCount > 0 ? digitCount - other.digitCount :
other.digitCount > 0 ? -1 :
digitCount > 0 ? 1 :
number - other.number;
}
}
protected static int digitCount(int digit, String number) {
char asChar = Character.forDigit(digit, 10);
int count = 0;
for (char c : number.toCharArray())
if (c == asChar)
++count;
return count;
}
protected static int digitCount(int digit, int number) {
number = Math.abs(number);
int count = 0;
while (number != 0) {
if (number % 10 == digit)
++count;
number /= 10;
}
return count;
}
public interface ToComparable<T, C extends Comparable<? super C>> {
C toComparable(T t);
}
public static <T, C extends Comparable<? super C>> void sort(List<T> list, ToComparable<T, C> function) {
class Pair implements Comparable<Pair> {
final T original;
final C comparable;
Pair(T original, C comparable) {
this.original = original;
this.comparable = comparable;
}
@Override
public int compareTo(Pair other) {
return
comparable == null && other.comparable == null ? 0 :
comparable == null ? -1 :
other.comparable == null ? 1 :
comparable.compareTo(other.comparable);
}
}
List<Pair> pairs = new ArrayList<>(list.size());
for (T original : list)
pairs.add(new Pair(original, function.toComparable(original)));
Collections.sort(pairs);
ListIterator<T> iter = list.listIterator();
for (Pair pair : pairs) {
iter.next();
iter.set(pair.original);
}
}
}