1 題目

現(xiàn)在有一個(gè)包含n個(gè)物體的數(shù)組，其中物體顏色為顏色為紅色、白色或藍(lán)色，請(qǐng)對(duì)這個(gè)數(shù)組進(jìn)行排序，讓相同顏色的物體相鄰，顏色的順序?yàn)榧t色，白色，藍(lán)色。

我們用0,1,2分別代表顏色紅，白，藍(lán)

注意：

本題要求你不能使用排序庫(kù)函數(shù)

擴(kuò)展：

一個(gè)非常直接的解法是兩步的計(jì)數(shù)排序的算法

首先：遍歷一遍數(shù)組，記錄0,1,2的數(shù)量，然后重寫(xiě)這個(gè)數(shù)組，先將0寫(xiě)入，再將1寫(xiě)入，再將2寫(xiě)入

你能給出一個(gè)只用一步，并且能在常數(shù)級(jí)空間復(fù)雜度解決這個(gè)問(wèn)題的算法嗎？

2 解法

2.1 思路

剛開(kāi)始我的思路是左右交換，弄左右指針，從左開(kāi)始，遇到0 1就跳過(guò)，遇到2就讓右指針往左，同樣是遇到1 2跳過(guò)，遇到0后左右交換，但是沒(méi)考慮1應(yīng)該怎么辦。。。

2.2 參考答案

然后就看了答案，答案是搞三個(gè)指針，左指針標(biāo)0,右指針標(biāo)2,再搞一個(gè)1指針，讓1指針從左開(kāi)始，遇到0和左指針換，遇到2和右指針換，遇到1跳過(guò)。具體做法：

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 更深一步的思考（其實(shí)并沒(méi)有增加什么性能）

在上面的算法代碼run起來(lái)之后發(fā)現(xiàn)了這么個(gè)事兒：

因?yàn)樯厦娴倪壿嫗?#xff0c;如果中間指針找到了0,就跟0指針交換，然后倆人同時(shí)++，但是如果0指針下個(gè)數(shù)還是0,這樣就會(huì)造成下次中間指針找到0之后就是0跟0交換造成了一次運(yùn)算浪費(fèi)（右指針--后碰到2一個(gè)道理），所以多執(zhí)行了一個(gè)檢查：

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

這樣的話(huà)還要保證不能反著換（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;}}} };

然后發(fā)現(xiàn)其實(shí)并沒(méi)有增加什么性能。。說(shuō)明我忽略了主要矛盾，導(dǎo)致性能并沒(méi)有質(zhì)的提升。

總結(jié)

以上是生活随笔為你收集整理的leetcode 75 --- sort-colors的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。