荷兰国旗问题

题目一:
给定一个数组arr,和一个数num,请把小于等于num的数放在数组的左边,大于num的数放在数组的右边。
要求额外空间复杂度O(1),时间复杂度O(n)。

解题思路:定义一个less区域,这个区域里的数字一定小于等于num,所以less指针一开始指向数组的前一个位置上,cur指针指在当前数组的第一个位置上,如果cur指针的值小于less指针下一个指向的位置的话,说明这个cur指针指向的数字数据less区域,把两个指针指向的数交换,less指针向后移,反之,就不交换只把cur指针向后移动一位。

void Swap(int &a,int &b)
{
  if(a==b)return; a
=a^b; b=a^b; a=a^b; } void partition(int* &arr,int l,int r,int num) { int less=l-1; int cur=l; while(cur<=r) { if(arr[cur]<=num) { Swap(arr[cur++],arr[++less]); } else { cur++; } } }

 

 

问题2:给定一个数组arr,和一个数num,请把小于num的数放在数组的左边,等于num的数组的中间,大于num的数放在数组的右边。

要求额外空间复杂度O(1),时间复杂度O(n)

这个问题比上一个略复杂,需要两个区域:less和more分别存放在比num大的数字和和比num小的数字。

void partition(int* &arr,int l,int r,int num)
{
    int less=l-1;
    int more=r+1;
    int cur=l;

    while(cur<more)
    {
        if(arr[cur]<num)
        {
            Swap(arr[cur++],arr[++less]);
        }
        else if(arr[cur]==num)
        {
            cur++;
        }
        else 
        {
            Swap(arr[cur],arr[--more]);
        }

    }

}

 

posted @ 2020-04-23 11:43  肉松松松松  阅读(108)  评论(0)    收藏  举报