数组分组问题

剑决浮云气

于 2021-01-25 10:16:11 发布

阅读量184

点赞数 1

CC 4.0 BY-SA版权

分类专栏：排序算法刷题文章标签：数据结构算法 leetcode

本文链接：https://blog.youkuaiyun.com/ZZHinclude/article/details/113104151

刷题同时被 2 个专栏收录

16 篇文章

订阅专栏

排序算法

11 篇文章

订阅专栏

本文介绍了一种在O(N)时间复杂度和O(1)额外空间复杂度下将数组中小于给定数值的元素放置左侧、大于该数值的元素放置右侧的方法。通过递归方式实现，适用于需要对数组进行快速划分的场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

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

#include <iostream>
using namespace std; 

const int N = 100010;
int Arr[N];

void Sort(int *p, int nStart, int nEnd);

int main(int argc, char** argv) 
{
	int nLen;
	scanf("%d",&nLen);
	
	for(int i = 0; i < nLen; i++)
	{
		scanf("%d",&Arr[i]);
	}
	
	Sort(Arr, 0, nLen-1);
	
	for(int i = 0; i < nLen; i++)
	{
		printf("%d ",Arr[i]);
	}
	printf("\n");
	
	return 0;
}

void Sort(int *p, int nStart, int nEnd)
{
	int nMid = (nEnd-nStart)/2 + nStart;
	int nFlag = p[nMid];
	int l = nStart-1; int r = nEnd+1;
	
	while(l < r)
	{
		do
		{
			l++;
		}while(Arr[l] < nFlag);
		
		do
		{
			r--;
		}while(Arr[r] > nFlag);
		
		if(l < r)
		{
			swap(Arr[l], Arr[r]);
		}
	}
}