求数组中最大n个数和最小n个数的和

发表于 分类于 ， 阅读次数： Waline： 本文字数： 1.3k 阅读时长 ≈ 1 分钟

题目描述

输入一个数 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

思路：

第一步先排序去重。然后判断此时数组中的元素是否大于 2 * N，如果不大于直接输出-1，然后用双指针从头和尾同时遍历相加，当left == N - 1时结束。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include <bits/stdc++.h>
#include <unordered_map>
using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef vector<int> vi;


int main(){
    int n;
    cin >> n;
    vector<int> nums(n);
    for(int i = 0; i < n; i++){
        cin >> nums[i];
    }
    sort(nums.begin(), nums.end());
    nums.erase(unique(nums.begin(), nums.end()), nums.end());
    int N;
    cin >> N;
    if(2* N > nums.size()){
        cout << -1 << endl;
        return 0;
    }
    //双指针
    int left = 0, right = nums.size() - 1;
    int res = 0;
    while(left < right){
        if(left < N){
            res += nums[left] + nums[right];
            if(left == N - 1){
                cout << res << endl;
                break;
            }
            left++;
            right--;
        }
    }
    return 0;
}