1 題目

現在有一個包含n個物體的數組，其中物體顏色為顏色為紅色、白色或藍色，請對這個數組進行排序，讓相同顏色的物體相鄰，顏色的順序為紅色，白色，藍色。

我們用0,1,2分別代表顏色紅，白，藍

注意：

本題要求你不能使用排序庫函數

擴展：

一個非常直接的解法是兩步的計數排序的算法

首先：遍歷一遍數組，記錄0,1,2的數量，然后重寫這個數組，先將0寫入，再將1寫入，再將2寫入

你能給出一個只用一步，并且能在常數級空間復雜度解決這個問題的算法嗎？

2 解法

2.1 思路

剛開始我的思路是左右交換，弄左右指針，從左開始，遇到0 1就跳過，遇到2就讓右指針往左，同樣是遇到1 2跳過，遇到0后左右交換，但是沒考慮1應該怎么辦。。。

2.2 參考答案

然后就看了答案，答案是搞三個指針，左指針標0,右指針標2,再搞一個1指針，讓1指針從左開始，遇到0和左指針換，遇到2和右指針換，遇到1跳過。具體做法：

class Solution { public:void sortColors(int A[], int n) {int zeroindex = 0;int twoindex = n - 1;int i = 0;while(i < twoindex + 1){if(A[i] == 0){swap(A[i],A[zeroindex]);zeroindex++;i++;}else if(A[i]==2){swap(A[i],A[twoindex]);twoindex--;}elsei++;}} };

2.3 更深一步的思考（其實并沒有增加什么性能）

在上面的算法代碼run起來之后發現了這么個事兒：

因為上面的邏輯為，如果中間指針找到了0,就跟0指針交換，然后倆人同時++，但是如果0指針下個數還是0,這樣就會造成下次中間指針找到0之后就是0跟0交換造成了一次運算浪費（右指針--后碰到2一個道理），所以多執行了一個檢查：

0指針（2指針）跟中間指針交換完之后，當當前所在數為0（2），就++（--）到非0（2）為止。

這樣的話還要保證不能反著換（0指針跑到1指針前面去了），所以有：

class Solution { public:void sortColors(int A[], int n) {int l = 0;int r = n - 1;int iter = 0;while (iter <= r) {if (A[iter] == 0) {swap(A[iter], A[l]);while (A[l] == 0)l ++;if (iter < l)iter = l;continue;}if (A[iter] == 1) {iter ++;continue;}if (A[iter] == 2) {swap(A[iter], A[r]);while (A[r] == 2)r --;continue;}}} };

然后發現其實并沒有增加什么性能。。說明我忽略了主要矛盾，導致性能并沒有質的提升。

總結

以上是生活随笔為你收集整理的leetcode 75 --- sort-colors的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。