前言:这是我们大学社团给我们大一新生布置的每日一题任务,因为本人编程理解有限,参考价值较低。
题目链接https://leetcode-cn.com/problems/distribute-candies/ 给定一段数列,思考元素种类的数学问题。在这里我们可以从题目中想到:假设我们的糖果数量为2n,有x种不同种类的糖果,那么我们会分两种情况去考虑:
①n<=x<=2n,不难看出答案为n
②0<x<=n,那这里的答案就是x
所以我们最后的答案就是min(n,x),我们的程序也就是只要寻找数组元素种类就行了。
我想到了前不久我们做过的一道快慢指针问题:删除有序数组中的重复项 ,在这一题中,我们删去了有序数组中的重复量,自然就可以得到元素的种类了,我在此文的结尾处写了这么一句话:
如果题目中是无序数组,我们还可以用<algorithm>库中的sort()函数给他排个序。
不就是这道题嘛awa,科普一下sort()函数的用法:
sort(数组名,数组长度,排序方式(可选));
从小到大排就是sort(数组名,数组长度);
那么我们的答案呼之欲出:先排序,再找元素种类,最后输出。但是我们在实际码代码的时候,突然发现:这个网站不给用algorithm库!
不过不要紧,我们可以当场给它写一个排序,像algorithm库中的sort快排是O(nlogn)时间复杂度的,而且相当相当难写,所以这里教一下比较容易但是时间复杂度是O(n^2)的冒泡排序。
下面放上冒泡排序的代码:
void swap(int& a,int& b){
int t=a;
a=b;
b=t;
return;
}
void sort(vector<int>& a,int L){
for(int i=0;i<L;i++){
for(int j=i;j>0;j--){
if(a[j]<a[j-1])swap(a[j],a[j-1]);
}
}
return;
}
打住!
这里的题目说数组长度过万,我们是不可以使用时间复杂度为O(n^2)的算法的!
所以我在这里提供另一个解题思路:用集合。
这么一说大家都能明白了,整一个200001大小的bool数组(看看数组中数值范围±100000,想想这里为什么应该是200001),遍历数组,出现新的数字就将这个boll数组中[新数字+100000]设置为true并将一个变量t++,如果boll本来就为true,说明这个数字出现过,跳过,最后t的值就是元素种类了。这里直接建bool数组的话就可以了,但是我提供一个空间复杂度不一样(O(nlongn)而不是O(数值范围))的方法:
众所周知,<map>这个东西非常的神奇,可以将两个类型的变量一一对应,而且用法是很直观的,可以当集合用,我就直接放出用mapAC的代码了:
class Solution {
public:
int distributeCandies(vector<int>& candyType) {
map<int,bool> m;
int L = candyType.size(),ans=0;
for(int i=0;i<L;i++){
if(m[candyType[i]]==false){
m[candyType[i]] = true;
ans++;
}
}
return min(ans,L/2);
}
};