求数组中最大N个数和最小N个数的和
题目描述
输入一个数 M ,表示数组中有 M 个数
输入 M 个数。
输入 n
求数组 M 中,去除重复值后,最大 n 个数和最小 n 个数的和
注意:最大和最小的数中不能有重复值,否则输出 -1
样例输入
5
3 3 2 4 2
2
样例输出
-1
说明
去除重复后最大的2个数为[4,3],最小的2个数为[2,3];有相同值,所以返回-1
样例输入
5
3 3 2 4 2
1
样例输出
6
说明
去除重复后最大的1个数为[4],最小的1个数为[2];没有相同值,返回6
```cpp
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
//双指针删除有序数组中重复元素
int removeDuplicates(vector<int>& nums,int length) {
int n = nums.size();
int slow = 1; int fast = 1;
while (fast < n) {
if (nums[fast] != nums[fast - 1]) {
nums[slow] = nums[fast];
slow++;
}
else length--;
fast++;
}
return length;
}
int main(){
int n, k;
vector<int> nums;
cin >> n;
for (int i = 0; i < n; i++) {
int x;
cin >> x;
nums.push_back(x);
}
cin >> k;
if (k <= 0) {
cout << -1 << endl;
}
sort(nums.begin(), nums.end());//先排序
int len = removeDuplicates(nums, n);
if (len < (2 * k) || len <= 0) {
cout << -1 << endl;
return -1;
}
int sum = 0;
for (int j = 0; j < k; j++) {//取前N个求和
sum += nums[j];
}
for (int j = len - 1; j >= len - k; j--) {//取后N个求和
sum += nums[j];
}
cout << sum << endl;
return sum;
}