写一个范围函数,可以传入开始和结束索引,然后返回整个列表。
function range(start, end) {
var ans = [];
for (let i = start; i <= end; i++) {
ans.push(i);
}
return ans;
}
当传入的开始索引和结束索引相同时会返回只包含一项数组。
function range(start, end) {
if (start === end) return [start];
// recursive case
}
结合上面的条件判断,再次优化函数。
function range(start, end) {
if (start === end) return [start];
return [start, ...range(start + 1, end)];
}
在我看来,比“循环”解决方案要优雅得多。但是如果我们使用新的Array(n)来创建一个包含n个元素的数组,我们甚至可以更进一步。
如果我们有一个n个元素列表,我们可以通过将每个元素映射到它的索引来构建一个范围。
function range(start, end) {
return (new Array(end - start + 1)).fill(undefined).map((_, i) => i + start);
}
在处理大范围时,实质上是构建一个大型数组。我们可以通过使用生成器来获得更高效的方法。
function* range(start, end) {
for (let i = start; i <= end; i++) {
yield i;
}
}
我们可以在循环中使用这个生成器(这将是非常有效的)。
for (i of range(1, 5)) {
console.log(i);
}
/* 1 2 3 4 5 */
[...range(1, 5)] // [1, 2, 3, 4, 5]
结合上面的各种条件,我们的最后代码如下:
function* range(start, end) {
yield start;
if (start === end) return;
yield* range(start + 1, end);
}