九章算法面试题69 排颜色 sort colors

九章算法官网-原文网址

http://www.jiuzhang.com/problem/70/

题目

给一个数组，并且数组里面元素的值只可能是0,1,2，然后现在把这个数组排序。

在线测试本题

http://www.lintcode.com/en/problem/sort-colors/

解答

这道题one pass的方法需要用two pointer的技巧。 用两个指针，一个指针x指向数组头便是0和1的区分的位置， 另一个指针y指向尾表示1和2的区分的位置，然后我们在遍历一遍数组，如果当前遇到的元素是0，那么我们就把它和x指向位置的元素调换，并且x前进一步，更新边界；如果是1，那么不变；如果是2，那么把它y指向的位置的元素调换，并且y后退一步，这样就可以保证x是0和1区分的位置，y是1和2区分的位置。 这种方法时间复杂度O(n)，空间复杂度O(1)