考研数据结构(每日一题)
题目:荷兰国旗问题:设有一个仅有红、白、蓝三种颜色的条块组成的条块序列,请编写一个时间复杂度为O(n)的算法,使得这些条块按红、白、蓝的顺序排好,即排成荷兰国旗图案。
算法思想:
顺序扫描线性表,将红色条块交换至线性表的最前面,蓝色条块交换至线性表的最后面。
设置三个指针,j:工作指针,表示当前扫描的元素,i以前的元素全部为红色,k以后的元素全部为蓝色。根据j所指示元素的颜色,决定将其交换至前部或尾部。初识时i=0,k=n-1
完整代码:
typedef enum {
RED,WHITE,BLUE
}color; //设置枚举数组
void Flag_Arrange(color a[],int n){
int i = 0,j = 0,k = n - 1;
while (j <= k)
{
switch (a[j]) //判断条块的颜色
{
case RED:Swap(a[i],a[j]);
//红色,则和i交换
i ++;
j ++;
break;
case WHITE:
j ++;
break;
case BLUE:Swap(a[j],a[k]);
k --;
//蓝色,则和k交换
//没有写j++是防止交换后a[j]仍为蓝色的情况
}
}
}