Java根据另一个列表对列表进行排序

时间:2013-03-22 09:44:14

标签: java sorting artificial-intelligence

我有一个以下类型的ArrayList:

class Move
{
    int from, to;
}

from属性始终具有值。如果未设置,则to属性将为-1。我有以下数组:

int[][] history = new int[50][50];

其中维度对应于移动类的“from”和“to”。在我的搜索功能中,根据我需要做的某些条件:

List<move> moves = board.getMoves();
for (int i = 0; i < moves.size(); i++)
    history[move.from][move.to]++;

因为move.to也可以是-1,我应该增加2d数组1的维度然后执行:

history[move.from+1][move.to+]++;

另外,根据上面的移动列表和历史数组,我需要根据相应历史索引的计数器按降序对移动列表进行排序。

这可能吗?

3 个答案:

答案 0 :(得分:1)

您可以使用Collections.sort(List, Comparator)执行Comparator,它会按您的意愿排序。

答案 1 :(得分:0)

是的,你可以使你的比较器使用历史数组。例如,我根据另一个数组counts对我的int列表进行排序。

public static void main(String[] args) {
    List<Integer> list = new ArrayList<>();
    list.addAll(Arrays.asList(new Integer[]{0, 1, 2, 3, 4, 5}));
    final int[] counts = new int[] {3, 4, 1, 7, 0, 1};

    Collections.sort(list, new Comparator<Integer>() {

        @Override
        public int compare(Integer arg0, Integer arg1) {
            return counts[arg1] - counts[arg0];
        }
    });

    System.out.println(list);
}

输出:[3, 1, 0, 2, 5, 4]

您的compare会是这样的:

@Override
public int compare(Move move0, Move move2) {
    return history[move1.from+1][move1.to] - history[move0.from+1][move0.to];
}

答案 2 :(得分:0)

您可以将历史记录设置为HashMap或单独的类,以使其更容易。但是因为你也希望能够根据频率对历史进行排序,我建议使用History类:

class Move {

   int from, to;

   @Override
   public int hashCode() {
      return from + (to * 100);
   }

   @Override
   public boolean equals(Object o) {
      return (o instanceof Move
              && ((Move) o).from == from
              && ((Move) o).to == to);
   }
}

class History extends Move implements Comparable<History> {

   int frequency;

   public History(Move m) {
      from = m.from;
      to = m.to;
      frequency = 1;
   }

   public void increment() {
      frequency += 1;
   }

   public int compareTo(History h) {
      // to be able to sort it in a TreeSet descending on frequency
      // note that it is not resorted if you change frequencies, so 
      // build the set, and then convert it to a TreeSet afterwards.
      return (frequency == h.frequency) ? 1 : (h.frequency - frequency);
   }
}

然后创建一个HashMap以快速填充历史记录,并将其转换为TreeSet进行排序:

  List<Move> moves = board.getMoves();
  HashMap<History, History> fillTable = new HashMap<History, History>();
  for (Move m : moves) {
     History h = fillTable.get(m);
     if (h == null) {
        h = new History(m);
        fillTable.put(h, h);
     } else {
        h.increment();
     }
  }
  TreeSet<History> sorted = new TreeSet<History>(fillTable.values());
  .... ready to use