荷兰国旗问题
荷兰国旗问题: 给定一个数组arr,和一个数num,请把小于num的数放在数组的左边,等于num的数放在数组的中间,大于num的数放在数组的右边。
要求 额外空间复杂度O(1),时间复杂度O(N)
解题思路:假设把数组分成三部分,小于num的区域,等于num的区域,大于num的区域。小于num的区域右边的前一个下标记为Less,等于num的区域最右边的下标记为cur,大于num的区域的左边的前一个下标记为more。
数组从0到N-1进行遍历,若当前元素小于num,则当前元素与小于num的区域最右边的数即Less所指的数进行交换,less++,cur++。若当前元素等于num,则当前元素不变,cur++。若当前元素大于num,则当前元素与大于num区域的左边一个元素(这里为待定区域)进行交换,并对交换过来的数与num进行比较。轮流下来依次比较,当cur与more等于时,停止。
代码如下:
# include <stdio.h>
void swap(int arr[], int i, int j) //数组arr中第i个元素与第j个元素进行交换
{
int swap_temp = arr[i];
arr[i] = arr[j];
arr[j] = swap_temp;
}
void partition(int arr[], int left, int right, int num)
{
int less = left - 1;
int more = right +1;
int cur = left;
int i = 0;
while( cur < more) //结束条件是cur与more相等
{
if(arr[cur] < num) //当前元素小于num,与小于区域交换
{
swap(arr, ++less, cur++);
}
else if(arr[cur] > more) //当前元素大于num,与大于区域交换并再次比较
{
swap(arr, --more, cur++);
}
else
{
cur++;
}
}
for(i = 0; i < (right - left + 1 ); i++)
{
printf("%d\t",arr[i]);
}
}
int main(void)
{
int a[] = {500, 22, 53, -158, 4, 86, 32, 33, 26, -70, 135, -34, 32, 543, 2500};
// int a[] = {1,3,4,2,5,6};
int n; //存放数组a中元素的个数
n = sizeof(a) / sizeof(a[0]); /*a[0]是int型, 占4字节*/
partition(a,0,n-1,100);
printf("\r\n");
return 0;
}