在算法题中,sort函数自定义排序,是经常使用的。在C++中通常这样写
sort(nums.begin(), nums.end(), cmp);
其中cmp是一个函数指针,传入一个自定义比较函数,这个函数这样写
bool cmp(const int& a, const int& b)
{
return a > b; //从大到小排序
}
这个函数参数与容器的参数相对应,这个写法是一种简单直接的写法。
这种写法可以用lammda表达式简写
vector<Interval> res;
sort(ins.begin(), ins.end(), [](Interval a, Interval b){return a.start < b.start;});
在Python中
有sorted函数和.sort方法,其中有一个key参数,用来指定排序方法
key使用lambda表达式
student_tuples = [
('john', 'A', 15),
('jane', 'B', 12),
('dave', 'B', 10),
]
sorted(student_tuples, key=lambda student: student[2]) # sort by age
[('dave', 'B', 10), ('jane', 'B', 12), ('john', 'A', 15)]
在Java中,使用Arrays.sort方法,使用匿名内部类,实现Comparator接口,重写compare方法即可。
也可以更改为lambda表达式
import java.util.*;
public class Main {
public static void main(String[] args){
Integer[] arr = {5,4,7,9,2,12,54,21,1};
//降序
Arrays.sort(arr, new Comparator<Integer>() {
public int compare(Integer a, Integer b) {
return b-a;
}
});
System.out.println(Arrays.toString(arr));
}
}