这道题目有一个我觉得非常好的解法,即用三个变量i, j, k分别统计0出现次数,0,1出现次数,0,1,2出现次数,这样即可对A[]进行赋值。赋值顺序有讲究,即先赋2,再赋1,最后赋0,这样能够保证A[]对应位置填入正确的值。我觉得这种解法好的地方在于通用性强,因为对于有多于3个的数的这种排序,都能用类似办法很方便地进行解决,假如用普通的双指针来做,那将是特别麻烦的事情。下面贴上AC代码:
class Solution { public: void sortColors(int A[], int n) { if (A == NULL || n <= 1) return; int i = -1, j = -1, k = -1; for (int idx = 0; idx < n; idx++) { if (A[idx] == 0) { A[++k] = 2; A[++j] = 1; A[++i] = 0; } else if (A[idx] == 1) { A[++k] = 2; A[++j] = 1; } else { A[++k] = 2; } } } };
原文:http://blog.csdn.net/hevc_cjl/article/details/38335055