Java中按照不同字段进行排序

发布于:2025-03-17 ⋅ 阅读:(22) ⋅ 点赞:(0)

在Java中(默认Java8),假设有一个类定义为:

class Pair{
	int a,b;
}

现在有一个Pair列表pairs,其类型为ArrayList<Pair>.
问题:(1)按照a字段从大到小排序并输出
(2)按照b字段从小到大排序并输出
说明:你可以对类进行一定改造,也可以借助库函数、库工具等进行实现。
此外,代码尽可能保持简洁,不要有太多冗余

1. 快速开发示例:Comparator+Lambda表达式

import java.util.*;

class Pair{
	int a, b;
	public Pair(int a, int b){
		this.a = a;
		this.b = b;
	}
	// 方便输出观察
	@override
	public String toString(){
		return String.format("(%d, %d)", a, b);
	}
}

public class Main{
	public List<Pair> getPairs(){
		List<Pair> pairs = new ArrayList<>();
		// ...
	}
	public static void main(String[] args){
		// 构造一个 pairs 实例
		List<Pair> pairs = getPairs();
		// 按照字段 a 降序排列, 实现降序只需加负号即可
		pairs.sort(Comparator.compareInt(p->-p.a));
		System.out.println(pairs);
		pairs.sort(Comparator.compareInt(p->p.b));
		System.out.println(pairs);
	}
}

如上述示例所示,通过调用集合框架的sort方法并传入Comparator即可实现按照指定字段的排序,对于int类型,已经有一个Comparator.compareInt可以直接使用,内部使用Lambda表达式指定字段,对于需要降序输出的,添加负号即可。

2. 标准开发示例:Comparator+Getter

通常,按照JavaBean的设计思想,类的字段保持私有,如必要还需要设置为final类型以保证线程安全,对外暴露公共的getter方法,此时代码编写更加规范:

import java.util.*;
class Pair{
	private final int a, b;
	public Pair(int a, int b){
		this.a=a;
		this.b=b;
	}
	public int getA(){return a;}
	public int getB(){return b;}
	public String toString(){return String.format("(%d,%d)", a, b);}
}
public class Main{
	public List<Pair> getPairs(){
		List<Pair> pairs = new ArrayList<>();
		// ...
	}
	public static void main(String[] args){
		// 构造一个 pairs 实例
		List<Pair> pairs = getPairs();
		// 按照字段 a 降序排列, Pair::getA表达式+reversed方法
		pairs.sort(Comparator.compareInt(Pair::getA).reversed());
		System.out.println(pairs);
		// 按照字段 b 升序排列
		pairs.sort(Comparator.compareInt(Pair::getB));
		System.out.println(pairs);
	}
}

3. 若字段a,b为String或者其他的自定义类型(已实现Comparable)

使用更通用的 comparing 方法

// 按照 a 降序排列
pairs.sort(Comparator.comparing(Pair::getA).reversed());

// 按照 b 升序排列
pairs.sort(Comparator.comparing(Pair::getB));

```

此处看到getter的好处,在a,b类型变化时,排序代码基本保持不变。


### 4. 如何实现 Comparable
假设现在需求并不是通过不同的字段来进行灵活多样的排序,而是结合不同字段实现更复杂的排序行为,此时就需要使用 Comparable 接口,例如通过a+b实现升序排序,或者先通过a进行升序排序,当a相等时通过b升序排序等等,以后者为例,实现代码如下:
```java
import java.util.*;

class Pair implements Comparable<Pair>{
	int a, b;
	public Pair(int a, int b){
		this.a = a;
		this.b = b;
	}
	public int compareTo(Pair o){
		if (this.a==o.a){
			return this.b-o.b;
		}
		return this.a-o.a;
	}
}
// 排序代码
Collections.sort(pairs);
```