题目一:
给定一个数组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]); } } }
原文:https://www.cnblogs.com/RainzzZ/p/12759714.html