先来看一道题目:
明明想在学校中请一些同学一起做一项问卷调查,为了实验的客观性,他先用计算机生成了N个1到1000之间的随机整数(N≤100),对于其中重复的数字,只保留一个,把其余相同的数去掉,不同的数对应着不同的学生的学号。然后再把这些数从小到大排序,按照排好的顺序去找同学做调查。请你协助明明完成“去重”与“排序”的工作。
首先,这道题的数据并不大,所以前面介绍过的计数排序仍然可以使用,不过这次我们介绍一下STL中sort与unique的用法。
sort ( a , a+n, cmp) : 对a数组从a[0]到a[n-1]进行排序。cmp是指自定义排序函数,如果是从小到大排序则可以省略。
unique ( a , a+n) : 对a数组从a[0]到a[n-1]进行去重,要求数组已经有序,返回去重后最后一个元素对应的指针。(想要获得下标直接减去数组名即可)
如果没有unique,我们只能在排序后遍历原数组,还得再开新数组,就像这样:
for(int i=0;i<n;i++){if(a[i]!=a[i-1])s.push_back(a[i]);}
而如果使用unique,就可以直接获得去重后的数组,代码如下:
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e5;
int a[N];
int n;
int main()
{cin>>n;for(int i=0;i<n;i++){cin>>a[i];}sort(a,a+n); int len=unique(a,a+n)-a;cout<<len<<endl;for(int i=0;i<len;i++){cout<<a[i]<<' ';}return 0;
}
不过unique的去重并不是删除重复元素,只是把它们放在了返回值之后,所以我们输出时截止到unique返回的下标即可。
另外,如果要用sort进行从大到小的排序,我们只需定义一个cmp函数即可:
bool cmp(int a,int b)
{return a>b;
}