题目大意是给一个长度为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; }