1、问题分析
题目链接:https://leetcode-cn.com/problems/sort-colors/
本质上就是一个排序问题。代码我已经进行了详细的注释,理解应该没有问题,读者可以作为参考,如果看不懂(可以多看几遍),欢迎留言哦!我看到会解答一下。
2、问题解决
笔者以C++
方式解决。
#include "iostream"
using namespace std;
#include "algorithm"
#include "vector"
#include "queue"
#include "set"
#include "map"
#include "string"
#include "stack"
class Solution {
public:
void sortColors(vector<int> &nums) {
// 如果数组为空,直接返回即可
if (nums.empty()) {
return;
}
// 使用自定义快排
quickSOrt(nums, 0, nums.size() - 1);
}
void quickSOrt(vector<int> &nums, int left, int right) {
if (left < right) {
int pos = partition(nums, left, right);
quickSOrt(nums, left, pos - 1);
quickSOrt(nums, pos + 1, right);
}
}
int partition(vector<int> &nums, int left, int right) {
int temp = nums[left];
while (left < right) {
while (left < right && nums[right] > temp) {
right--;
}
nums[left] = nums[right];
while (left < right && nums[left] <= nums[right]) {
left++;
}
nums[right] = nums[left];
}
nums[left] = temp;
return left;
}
};
int main() {
vector<int> nums = {2, 0, 2, 1, 1, 0};
Solution *pSolution = new Solution;
pSolution->sortColors(nums);
for (int i = 0; i < nums.size(); ++i) {
cout << nums[i] << " ";
}
cout << endl;
system("pause");
return 0;
}
运行结果
有点菜,有时间再优化一下。
3、总结
难得有时间刷一波LeetCode
, 这次做一个系统的记录,等以后复习的时候可以有章可循,同时也期待各位读者给出的建议。算法真的是一个照妖镜,原来感觉自己也还行吧,但是算法分分钟教你做人。前人栽树,后人乘凉。在学习算法的过程中,看了前辈的成果,受益匪浅。
感谢各位前辈的辛勤付出,让我们少走了很多的弯路!
哪怕只有一个人从我的博客受益,我也知足了。
点个赞再走呗!欢迎留言哦!