题意:荷兰国旗问题,给一个数组[0,0,2,1,0],构造成[0,0,0,1,2]的形式,分成三块
https://leetcode.com/problems/sort-colors/description/
题解:
在任意时刻,i 左边的数都是 0,k 右边的数都是 2,而 i 到 j 之间的数都是 1。
想象有三个指针, i , j , k i, j, k i,j,k 维护 [ 0 , i ) [0,i) [0,i)为0,维护$[i,j)为1,[j, nums.size()]为2
想象有三个指针在动,i代表起始位置,k代表末尾位置,j遍历整个数组,移动j,当j的值指向的数字为0,的时候那么跟i交换,移动的过程中j >=i
cpp
class Solution {
public:
void sortColors(vector<int>& nums) {
for(int i = 0, j = 0, k = nums.size()-1; k >= j;) {
if(!nums[j]) {
swap(nums[i++], nums[j++]);
} else if( nums[j] == 2) {
swap(nums[j],nums[k--]);
} else j++;
}
}
};
时间复杂度 O ( n ) O(n) O(n)
空间复杂度 O ( 1 ) O(1) O(1)