https://leetcode.com/problems/sort-colors/
https://leetcode.com/problems/sort-colors/discuss/26472/Share-my-at-most-two-pass-constant-space-10-line-solution
https://leetcode.com/problems/sort-colors/discuss/26500/Four-different-solutions
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?
https://leetcode.com/problems/sort-colors/discuss/26472/Share-my-at-most-two-pass-constant-space-10-line-solution
The idea is to sweep all 0s to the left and all 2s to the right, then all 1s are left in the middle.
It is hard to define what is a "one-pass" solution but this algorithm is bounded by O(2n), meaning that at most each element will be seen and operated twice (in the case of all 0s). You may be able to write an algorithm which goes through the list only once, but each step requires multiple operations, leading the total operations larger than O(2n).
void sortColors(int A[], int n) {
int second=n-1, zero=0;
for (int i=0; i<=second; i++) {
while (A[i]==2 && i<second) swap(A[i], A[second--]);
while (A[i]==0 && i>zero) swap(A[i], A[zero++]);
}
}
public void sortColors(int[] nums) {
int zeroP=-1, twoP= nums.length;
for(int i=0; i<twoP; ){
if(nums[i]==0){
zeroP++;
int temp = nums[zeroP];
nums[zeroP]=0;
nums[i]=temp;
i++;
} else if(nums[i]==2){
twoP--;
int temp = nums[twoP];
nums[twoP]=2;
nums[i]=temp;
} else {
i++;
}
}
}
https://leetcode.com/problems/sort-colors/discuss/26500/Four-different-solutions
// two pass O(m+n) space
void sortColors(int A[], int n) {
int num0 = 0, num1 = 0, num2 = 0;
for(int i = 0; i < n; i++) {
if (A[i] == 0) ++num0;
else if (A[i] == 1) ++num1;
else if (A[i] == 2) ++num2;
}
for(int i = 0; i < num0; ++i) A[i] = 0;
for(int i = 0; i < num1; ++i) A[num0+i] = 1;
for(int i = 0; i < num2; ++i) A[num0+num1+i] = 2;
}
// one pass in place solution
void sortColors(int A[], int n) {
int n0 = -1, n1 = -1, n2 = -1;
for (int i = 0; i < n; ++i) {
if (A[i] == 0)
{
A[++n2] = 2; A[++n1] = 1; A[++n0] = 0;
}
else if (A[i] == 1)
{
A[++n2] = 2; A[++n1] = 1;
}
else if (A[i] == 2)
{
A[++n2] = 2;
}
}
}
// one pass in place solution
void sortColors(int A[], int n) {
int j = 0, k = n - 1;
for (int i = 0; i <= k; ++i){
if (A[i] == 0 && i != j)
swap(A[i--], A[j++]);
else if (A[i] == 2 && i != k)
swap(A[i--], A[k--]);
}
}
// one pass in place solution
void sortColors(int A[], int n) {
int j = 0, k = n-1;
for (int i=0; i <= k; i++) {
if (A[i] == 0)
swap(A[i], A[j++]);
else if (A[i] == 2)
swap(A[i--], A[k--]);
}
}