(新A卷,100分)- 最大N个数与最小N个数的和（Java & JS & Python）华为OD

(新A卷,100分)- 最大N个数与最小N个数的和（Java & JS & Python）华为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。

题目解析

简单的逻辑题。主要应该是考察数组常用方法的使用，以及数组去重如何实现。

Java算法源码

import java.util.Arrays;
import java.util.HashSet;
import java.util.Scanner;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);

    int m = sc.nextInt();

    int[] arr = new int[m];
    for (int i = 0; i < m; i++) {
      arr[i] = sc.nextInt();
    }

    int n = sc.nextInt();

    System.out.println(getResult(m, arr, n));
  }

  public static int getResult(int m, int[] arr, int n) {
    HashSet<Integer> set = new HashSet<>();

    for (int val : arr) {
      if (val < 0 || val > 1000) return -1;
      set.add(val);
    }

    if (set.size() < n * 2) return -1;

    Integer[] distinct_arr = set.toArray(new Integer[0]);

    Arrays.sort(distinct_arr, (a, b) -> a - b);

    int l = 0;
    int r = distinct_arr.length - 1;
    int ans = 0;

    while (n > 0) {
      ans += distinct_arr[l] + distinct_arr[r];
      l++;
      r--;
      n--;
    }

    return ans;
  }
}

JavaScript算法源码

/* JavaScript Node ACM模式 控制台输入获取 */
const readline = require("readline");

const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout,
});

const lines = [];
rl.on("line", (line) => {
  lines.push(line);

  if (lines.length === 3) {
    const m = lines[0] - 0;
    const arr = lines[1].split(" ").map((ele) => Number(ele));
    const n = lines[2] - 0;
    console.log(getResult(m, arr, n));
    lines.length = 0;
  }
});

function getResult(m, arr, n) {
  const set = new Set();

  for (let val of arr) {
    if (val < 0 || val > 1000) return -1;
    set.add(val);
  }

  if (set.size < n * 2) {
    return -1;
  }

  arr = [...set].sort((a, b) => a - b);

  let l = 0;
  let r = arr.length - 1;
  let ans = 0;

  while (n > 0) {
    ans += arr[l] + arr[r];
    l++;
    r--;
    n--;
  }

  return ans;
}

Python算法源码

# 输入获取
m = int(input())
arr = list(map(int, input().split()))
n = int(input())


# 算法入口
def getResult(m, arr, n):
    # 你需要对数组进行去重
    arr = list(set(arr))

    # 最大N个数与最小N个数不能有重叠，如有重叠，输入非法返回-1
    if len(arr) < n * 2:
        return -1

    arr.sort()

    # 数组中数字范围[0, 1000]
    if arr[0] < 0 or arr[-1] > 1000:
        return -1

    # 最大N个数与最小N个数的和
    return sum(arr[:n]) + sum(arr[-n:])


# 算法调用
print(getResult(m, arr, n))