LeetCode - Refresh - Sort Colors

转载于 2015-03-23 15:11:00 发布 · 54 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/shuashuashua/p/4359863.html

本文介绍了一个高效的三色排序算法实现，该算法通过一次遍历将数组中的0、1、2三种元素分别放置在数组的不同位置，同时保持原有的顺序。算法采用三个指针：start、end 和 runner 进行操作，并利用 C++ 的 lambda 函数进行元素交换。

 1 class Solution {
 2 public:
 3     void sortColors(int A[], int n) {
 4         auto swap = [](int &a, int &b){int t = a; a = b; b = t;};
 5         int start = 0, end = n-1, runner = 0;
 6         while (runner <= end) {
 7             if (A[runner] == 0) {
 8                 swap(A[runner++], A[start++]);
 9                 continue;
10             }
11             if (A[runner] == 2) {
12                 swap(A[runner], A[end--]);
13                 continue;
14             }
15             runner++;
16         }
17     }
18 };