重新学习 Array.prototype.sort() 方法

小半 • 2024年3月19日下午9:34 • 技术漫谈 • 阅读 58

面试时被问到数组的排序，回答的不好，这里重新复习一下，列几条注意的点。

1. 排序时 `a` 、`b` 分别是什么。

对于：arr.sort((a,b)=>...)，我从来没记住 a 和 b 哪个是数组中前一个元素，哪个是后一个。。

实验看看：

const arr = [1, 3];
arr.sort((a,b)=>console.log(a, b));

在 Chrome console 中运行，输出如下：

现在知道了，在排序时，是拿后一个元素和前一个比较，所以 a 是后一个元素，b 是前一个元素。

2. 数组中有 `undefined` 时会发生什么？

先看这段代码：

const arr = [1, undefined, 3, undefined];
arr.sort();

它会使 arr 变成 [1, 3, undefined, undefined]，说明 undefined 默认排在最后。

再看看这段代码：

const arr = [1, undefined, 3, undefined];
arr.sort((a,b)=>console.log(a, b));

得到如下结果：

说明当比较的元素是 undefined 时，不会再调用排序方法( sort 的入参)，而是直接将 undefined 排在后面。

3. `sort()` 不传参时的注意事项

对于：

const months = ['March', 'Jan', 'Feb', 'Dec'];
months.sort();

我们得到的是 ["Dec", "Feb", "Jan", "March"]，这是因为 sort() 方法不传参时，默认按 UTF-16 code units values 进行排序(类似于我们常说的字典顺序)。

当数组中的元素不是字符串时，会先被转换成字符串再排序(undefined值除外)。(重要！)所以不要以为 [1, 2, 11, 22].sort() 会得到 [1, 2, 11, 22]，实际你得到的是 [1, 11, 2, 22]，这是因为它们是按照 ['1', '2', '11', '22'] 进行排序的。

4. 优化排序性能

在对一些数组进行排序时，可能涉及属性访问(方法调用)：

users.sort((a,b)=>a.f() - b.f());

// or
users.sort((a,b)=>f(a) - f(b));

而在排序过程中，同一个元素可能会被比较多次，所以方法调用也会有多次。

如果该方法调用比较重(耗性能)，可以用先 map 再 sort 的方式进行优化：

// 假设 user.f() 方法比较重

// 相当于对 v.f() 进行缓存，防止重复调用
const mapped = users.map((v,i)=>({
    i,
    value: v.f()
}));

// 对缓存的值进行排序
mapped.sort((a,b)=>a.value - b.value);

// 取真正的结果
const result = mapped.map(v=>users[v.i]);