题目大意是给一个长度为n的整数序列,然后再给一个数num,然后将小于这个num的放在序列前面,等于num的放在序列中间,大于num的放在序列的后面,不要求排序,时间复杂度O(n),空间复杂度O(1)(即不能构造其他数组)
这里没怎么用到算法的基础,而是运用数据的调配,首先我们在数组的两端放两个指针,一个head,一个end,head前面的是小于num区的数,end后面的是大于num区的数,中间的就是等于num区的数。
然后遍历数组,如果有小于num的数,就和小于区的下一个元素交换,然后小于区扩大一个,如果有大于num的数,就和大于区的前一个元素进行交换,然后大于区扩大一个,直到循环遍历到大于区为止
然后贴上代码
#include<cstdio>
#include<iostream>
#include <algorithm>
using namespace std;
int main()
{
int n,m;
int a[1000];
int num;
cin>>n>>num;
for(int i=0;i<n;i++) cin>>a[i];
int head=0,end=n-1;
for(int i=0;i<=end;){
if(a[i]<num){
swap(a[i],a[head++]);
i++;
}
else if(a[i]>num){
swap(a[i],a[end--]);
}
else i++;
}
for(int i=0;i<n;i++) cout<<a[i]<<' ';
return 0;
}