輸入乙個整型陣列,每個元素在0~2
之間,其中0,
1,2分別代表紅、白、藍。現要求對陣列進行排序,相同顏色的在一起,而且按紅白藍順序先後排列。要求時間複雜度為
o(n)
。最容易想到的是排序,比如快排,歸併,堆排等,但它們的時間複雜度為o(nlogn)
,與題意不符。
第二種想到的是計數排序,掃瞄一遍過去,分別記錄0,1
,2的個數,然後再對陣列進行賦值。時間複雜度為
o(2n)
,即o(n)
,滿足條件。
還有一種方法,只需掃瞄一遍陣列即可,其充分利用了元素只有3
種的特性:在掃瞄陣列的時候,使用首尾倆個指標,分別指示0、
1與1、
2邊界。比如源陣列為。
第一步:首指標p0,尾指標p1,
i標識當前掃瞄位置,當前位置值為
2,需要將其交換至尾指標p1
位置,p1要向前移動一位,p0、
i位置不變。
第二步:i位置值不為0、
2,i要向後移動一位,p0、
p1位置不變。
第三步:i
位置值為2,需要將其交換至尾指標p1
位置,並且
p1往前移動一位,i與
p0位置不變。
第四步:i
位置不為0、
2,i要向後移動一位,p0、
p1位置不變。
第五步:i
位置值為
0,需要將其交換至首指標
p0位置,並且
p0往後移動一位,i與
p1位置不變。
第六步:i
位置不為0、
2,i要向後移動一位,p0、
p1位置不變。
第七步:i
位置值為
0,需要將其交換至首指標
p0位置,並且
p0往後移動一位,i與
p1位置不變。
第八步:i
位置不為0、
2,i要向後移動一位,p0、
p1位置不變。
第九步:i
位置超過
p1位置了,結束。
#include using namespace std;
void threecolorsort( int narray, int ncount )
//當前值為0,與首指標p0指向的值相互交換,p0向後移動一位
//i、p0位置不變
else if ( narray[i] == 0 && i > p0 )
//i位置不為0、2,i要向後移動一位,p0、p1位置不變。
else
}}//書上的例子**
void sortcolors( int narray, int ncount )
else if ( narray[i] == 2 )
}}int main()
; //int narray = ;
//int narray = ;
//int narray = ;
int narray = ;
threecolorsort( narray, _countof(narray) );
//sortcolors( narray, _countof(narray) );
for( int i = 0; i < _countof(narray); ++i )
cout << endl;
return 0;
}
系列文章說明:
1.本系列文章[演算法練習],僅僅是本人學習過程的乙個記錄以及自我激勵,沒有什麼說教的意思。如果能給讀者帶來些許知識及感悟,那是我的榮幸。
2.本系列文章是本人學習陳東鋒老師《進軍矽谷,程式設計師面試揭秘》一書而寫的一些心得體會,文章大多數觀點均來自此書,特此說明!
3.文章之中,難免有諸多的錯誤與不足,歡迎讀者批評指正,謝謝.
java演算法之 三色棋
package demo algorithm gossip 三色棋 三色旗的問題最早由e.w.dijkstra所提出,他所使用的用語為dutch nation flag dijkstra為荷蘭 人 而多數的作者則使用three color flag來稱之。假設有一條繩子,上面有紅 白 藍三種顏色的旗...
基礎演算法之三色旗 python
問題 假設有一條繩子,上面有紅,白,藍三種顏色的旗子,起初繩子上的旗子顏色並沒有順序,您希望將之分類,並排列為藍,白,紅的順序,要如何移動次數才會最少,注意您只能在繩子上進行這份動作,而且一次只能調換兩個色旗子。方法 一開始b,w指向0索引,r指向最後乙個索引 1 如果w指向藍色 則交換b和w指向的...
nowcoder 三色排序
有乙個只由0,1,2三種元素構成的整數陣列,請使用交換 原地排序而不是使用計數進行排序。給定乙個只含0,1,2的整數陣列a及它的大小,請返回排序後的陣列。保證陣列大小小於等於500。測試樣例 0,1,1,0,2,2 6 返回 0,0,1,1,2,2 定義乙個左指標,乙個右指標。當遍歷到0時,和左指標...