题目地址: https://oj.leetcode.com/problems/sort-colors/
题目大致含义:一个数组有3个数字:0,1,2.将这个数组排序。0在最前面,2在最后面。 如果可以,尽量只扫描一遍数组。
method:
第一种想法:
扫描一遍数组,记录0,1,2的个数。第二遍分别写入对应个数的0,1,2.
代码如下:
class Solution {
public:
void sortColors(int A[], int n) {
int red=0, white=0, blue=0;
for(int i=0; i<n; i++)
{
if(A[i] == 0) red++;
if(A[i] == 1) white++;
if(A[i] == 2) blue++;
}
int pos = 0;
while(red-- >0)
{
A[pos++] = 0;
}
while(white-- >0)
{
A[pos++] = 1;
}
while(blue-- >0)
{
A[pos++] = 2;
}
}
};
如果只扫描一次数组的话。则需要对应额外的指针。
代码如下:
class Solution {
public:
void sortColors(int A[], int n) {
int i=-1, j=-1, k=-1;
for(int p = 0; p < A.length; p++)
{
if(A[p] == 0)
{
A[++k]=2;
A[++j]=1;
A[++i]=0;
}
else if (A[p] == 1)
{
A[++k]=2;
A[++j]=1;
}
else if (A[p] == 2)
{
A[++k]=2;
}
}
}
};
这个方法用i,j,k分别指向当前0,1,2的位置。但是这个方法存在的一个问题是:额外的写比较多。
下面的这个方法类似上面的方法,但是减少了额外的写。
代码:
class Solution {
public:
void sortColors(int A[], int n) {
int j = 0, k = n;
for(int i=0; i<k; i++)
{
if(A[i] == 0)
swap(A[j++], A[i]);
else if(A[i] == 2)
swap(A[--k], A[i--]);
}
}
};