华为OD机试统一考试D卷C卷 - 最大N个数与最小N个数的和（C++）

华为OD机考:OD统一考试D卷+C卷+A卷+B卷+刷题OJ

真题目录：华为OD机考机试 真题目录（ D卷 +C卷 + B卷 + A卷） + 考点说明
在线OJ：点击立即刷题，模拟真实机考环境
华为OD面试真题精选：华为OD面试真题精选

题目描述

给定一个数组，编写一个函数来计算它的最大N个数与最小N个数的和。你需要对数组进行去重。

说明：

• 数组中数字范围[0, 1000]
• 最大N个数与最小N个数不能有重叠，如有**重叠，输入非法**返回-1
• 输入非法返回-1

输入描述

• 第一行输入M， M标识数组大小
• 第二行输入M个数，标识数组内容
• 第三行输入N，N表达需要计算的最大、最小N个数

输出描述

输出最大N个数与最小N个数的和

用例

输入	5 95 88 83 64 100 2
输出	342
说明	最大2个数[100,95],最小2个数[83,64], 输出为342。

输入	5 3 2 3 4 2 2
输出	-1
说明	最大2个数[4,3],最小2个数[3,2], 有重叠输出为-1。

C++

#include <iostream>
#include <algorithm>
#include <unordered_set>
using namespace std;

int getSumOfMaxAndMinN(int size, int nums[], int n) {
    unordered_set<int> numSet;  // 使用无序集合去重

    // 将数组中的数字插入到集合中，同时判断数字是否符合要求
    for (int i = 0; i < size; i++) {
        if (nums[i] < 0 || nums[i] > 1000) return -1;  // 不符合要求，返回-1
        numSet.insert(nums[i]);
    }

    if (numSet.size() < n * 2) return -1;  // 数组中不足2n个不同的数字，返回-1

    int distinctNums[numSet.size()];  // 存储去重后的数字
    int index = 0;
    for (auto val : numSet) {
        distinctNums[index++] = val;
    }

    sort(distinctNums, distinctNums + numSet.size());  // 排序

    int left = 0;
    int right = numSet.size() - 1;
    int sum = 0;

    while (n > 0) {
        sum += distinctNums[left] + distinctNums[right];  // 计算最大和最小n个数字之和
        left++;
        right--;
        n--;
    }

    return sum;
}

int main() {
    int size;
    cin >> size;

    int nums[size];
    for (int i = 0; i < size; i++) {
        cin >> nums[i];
    }

    int n;
    cin >> n;

    cout << getSumOfMaxAndMinN(size, nums, n) << endl;

    return 0;
}