按自定义顺序按字段排序对象的ArrayList

时间:2016-04-08 08:32:53

标签: java sorting comparator

如何实现对字段名称内容的自定义排序:

  • 第一个元素:P后跟数字[1-9] {2}始终在第一个
  • 后跟:P后跟数字0 [0-9]
  • 其次是:S
  • 后跟数字[1-9] {2}
  • 然后其余按正常顺序i1.getName().compareToIgnoreCase(i2.getName())

    private static Comparator<Item> itemComperator = new Comparator<Item>() {
        @Override
        public int compare(Item i1, Item i2) {
            if (i1.getName().matches("P[1-9]{2}") && i2.getName().matches("P0[0-9]"))
                return -1;
            else if (i1.getName().matches("S[1-9]{2}"))
                return -1;
            else
                return i1.getName().compareToIgnoreCase(i2.getName());
        }
    };
    
    @Test
    public void sortItem() {
        Item item01 = new Item(1, "R59");
        Item item02 = new Item(2, "S48");
        Item item03 = new Item(3, "P01");
        Item item04 = new Item(4, "P25");
        Item item05 = new Item(5, "R99");
    
        List<Item> items = Arrays.asList(item01, item02, item03, item04, item05);
        System.out.println("before sorting");
        long seed = System.nanoTime();
        Collections.shuffle(items, new Random(seed));
        for (Item i : items) {
            System.out.println(i.getId() + " " + i.getName());
        }
    
        System.out.println("after sorting");
        Collections.sort(items, itemComperator);
        for (Item i : items) {
            System.out.println(i.getId() + " " + i.getName());
        }
    
    
    }
    
    public class Item {
        private int id;
        private String name;
    
        public Item(int id, String name) {
            this.id = id;
            this.name = name;
        }
    
        public int getId() {
            return id;
        }
    
        public void setId(int id) {
            this.id = id;
        }
    
        public String getName() {
            return name;
        }
    
        public void setName(String name) {
            this.name = name;
        }
    }
    

预期输出:

after sorting
4 P25
3 P01
2 S48
1 R59
5 R99

1 个答案:

答案 0 :(得分:5)

我认为我会首先将每个输入映射到&#34; kind&#34;数字,基于上面的标准列表。例如:

int kind(String input) {
  if (input.matches("P[1-9]{2}") {
    return 0;
  } else if (input.matches("P0[0-9]")) {
    return 1;
  } else if (input.matches("S.*")) {
    return 2;
  } else if (input.matches("[1-9]{2}")) {
    return 3;
  } else {
    return 4;
  }
}

这使您可以查看两个字符串是否相同&#34; kind&#34 ;;如果没有,请根据种类返回订购。如果它们是相同类型,只需使用(不区分大小写)词典排序进行比较(您没有指定,但我认为您需要例如&#34; P11&#34;来之前&#34; P22&#34 ):

public int compare(Item a, Item b) {
  String nameA = a.getName();
  String nameB = b.getName();

  int kindA = kind(nameA);
  int kindB = kind(nameB);
  if (kindA != kindB) {
    return Integer.compare(kindA, kindB);
  } else {
    return nameA.compareToIgnoreCase(nameB);
  }
}