给定一个包含红色、白色和蓝色,一共 n 个元素的数组,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。
此题中,我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。
注意:
不能使用代码库中的排序函数来解决这道题。
示例:
输入: [2,0,2,1,1,0]
输出: [0,0,1,1,2,2]
进阶:
一个直观的解决方案是使用计数排序的两趟扫描算法。
首先,迭代计算出0、1 和 2 元素的个数,然后按照0、1、2的排序,重写当前数组。
你能想出一个仅使用常数空间的一趟扫描算法吗?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sort-colors
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
其实就是排序,而且没有要求稳定性,就是用啥排序算法都行。题目并没有给出数据规模,因此我默认数据量不大,直接选择了冒泡排序
冒泡排序的时间复杂度是N平方,无法优化,但是可以进一步优化常数项, 比如循环的起止条件。由于每一次遍历都会将最后一位“就位”,因此内层循环的截止条件就可以是 nums.length-i
, 而不是 nums.length
, 可以省一半的时间。
/*
* @lc app=leetcode id=75 lang=javascript
*
* [75] Sort Colors
*
* https://leetcode.com/problems/sort-colors/description/
*
* algorithms
* Medium (41.41%)
* Total Accepted: 297K
* Total Submissions: 716.1K
* Testcase Example: '[2,0,2,1,1,0]'
*
* Given an array with n objects colored red, white or blue, sort them in-place
* so that objects of the same color are adjacent, with the colors in the order
* red, white and blue.
*
* Here, we will use the integers 0, 1, and 2 to represent the color red,
* white, and blue respectively.
*
* Note: You are not suppose to use the library's sort function for this
* problem.
*
* Example:
*
*
* Input: [2,0,2,1,1,0]
* Output: [0,0,1,1,2,2]
*
* Follow up:
*
*
* A rather straight forward solution is a two-pass algorithm using counting
* sort.
* First, iterate the array counting number of 0's, 1's, and 2's, then
* overwrite array with total number of 0's, then 1's and followed by 2's.
* Could you come up with a one-pass algorithm using only constant space?
*
*
*/
/**
* @param {number[]} nums
* @return {void} Do not return anything, modify nums in-place instead.
*/
var sortColors = function(nums) {
function swap(nums, i, j) {
const temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
for(let i = 0; i < nums.length - 1; i++) {
for(let j = 0; j < nums.length - i; j++) {
if(nums[j] < nums[j -1]) {
swap(nums, j - 1, j)
}
}
}
};