题目链接
LeetCode 1122: https://leetcode-cn.com/problems/relative-sort-array
首先我们一起来看题目:
解题思路
这是一道排序的题目,那么我们首先应该想到一些常见的排序算法。主要分为两类,一类是基于比较的排序算法,例如快速排序和堆排序等。另一类是非基于比较的排序算法,包括计数排序和桶排序等。基于比较的排序算法的最优化时间复杂度也无法小于 。而非基于比较的排序算法则可以实现更低的时间复杂度。
基于比较的排序算法应用场景更广泛,而非基于比较的排序算法有自己的局限性,只适用于有条件限制的场景,例如限制了元素大小等。而本题正是限制了数组长度和元素大小,因此联想到计数排序。
本题解不对计数排序进行详细介绍,如果你看不懂题解,可以去查找相关资料学习,但我会尽可能地将题解写得更简单易懂。
解法如下:
首先要全面理解题目的要求和意图,我们需要将 中的元素按照 的顺序进行排列,并且仅存在于 中的元素,需要按照从小到大的顺序放在 的末尾。
因为 中每个元素的出现次数不一定,所以我们首先遍历 ,统计其中每个元素的出现次数,并记录在 中, 中元素的下标为 中被统计元素的值, 是我们的一个中间数组。例如 中值为 的元素出现了 次,那么我们统计之后得到 。
接下来遍历 ,假设 中的元素为 ,那么如果
arr[num] !== 0
,则说明 中还有值为 的元素没有被排序,那么我们就可以ans.push(num)
,同时arr[num]--
。这样就完成了按照 中元素的顺序对 中的元素进行排序。最后遍历 ,将在 中,但不在 中的元素按照从小到大的顺序,附加到 的末尾,最后返回 即可得到排序后的数组。
根据题意可知,数组长度范围固定,所以可以声明固定长度的中间数组,我使用
new Int8Array(1001)
声明 型的数组。这里设定其长度为 是因为,数组 的下标与 中元素的值相对应,而其元素最大值为 。
代码
/**
* @param {number[]} arr1
* @param {number[]} arr2
* @return {number[]}
*/
var relativeSortArray = function(arr1, arr2) {
let arr = new Int8Array(1001);
let ans = [];
for (let i = 0; i < arr1.length; i++) {
arr[arr1[i]]++;
}
for (let num of arr2) {
while (arr[num] !== 0) {
ans.push(num);
arr[num]--;
}
}
for (let i = 0; i < arr.length; i++) {
while (arr[i] !== 0) {
ans.push(i);
arr[i]--;
}
}
return ans;
};
复杂度分析:
时间复杂度:
空间复杂度:
其中 为数组 的长度, 为数组 的长度。
更多题解请关注:https://github.com/leviding/leetcode-js-leviding
每天一道算法题,趣学算法很简单!
看完三件事
如果你觉得本文对你有帮助,我想请你帮个忙:
转发本文,点赞或者点个「在看」,是对我最大的认可和支持;
关注公众号「技术漫谈」,订阅更多精彩内容,获取更多学习资料;
公众号后台回复「加群」,加入算法和技术交流群,与更多读者交流。
长按上方二维码关注公众号「技术漫谈」,订阅更多精彩内容。
你“在看”我吗?