OJ3260最大数组和问题

题目:

小明是一名勇敢的冒险家,他在一次探险途中发现了一组神秘的宝石,这些宝石的价值都不同。但是,他发现这些宝石会随着时间的推移逐渐失去价值,因此他必须在规定的次数内对它们进行处理。 小明想要最大化这些宝石的总价值。他有两种处理方式:

1.选出两个最小的宝石,并将它们从宝石组中删除。

2.选出最大的宝石,并将其从宝石组中删除。 现在,给你小明手上的宝石组,请你告诉他在规定的次数内,最大化宝石的总价值是多少。

输入格式 第一行包含一个整数t,表示数据组数。 对于每组数据,第一行包含两个整数n和k,表示宝石的数量和规定的处理次数。

第二行包含n个整数a1,a2,...,an,表示每个宝石的价值。 输出格式 对于每组数据,输出一个整数,表示在规定的次数内,最大化宝石的总价值。

思路:

首先,对原始数组进行排序不会影响结果,因为最小宝石始终在数组的开头,最大宝石则在数组的未尾。也就是排序后,每次操作要么删除左边的两个元素,要么删除右边的一个元素。因此,如果我们删除2m 个最小宝石和(k-m)个最大宝石,则剩余的元素组成的段在排序后的数组中从位置(2m + 1)到位置(n-(k-m)),可以从左到右遍历 ,使用前缀和在 O(1)时间内计算其总和。

样例输入

cpp 复制代码
6
5 1
2 5 1 10 6
5 2
2 5 1 10 6
3 1
1 2 3
6 1
15 22 12 10 13 11
6 2
15 22 12 10 13 11
5 1
999999996 999999999 999999997 999999998 999999995

样例输出

cpp 复制代码
21
11
3
62
46
3999999986

代码:

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() {
    int t;
    cin >> t; // 读取测试用例的数量
    while(t--) { // 对每一个测试用例进行处理
        int n, k;
        cin >> n >> k; // 读取宝石的数量n和处理次数k
        vector<ll> a(n), sum(n + 1, 0); // 初始化数组a用于存放宝石价值,sum用于存放前缀和
        for(int i = 0; i < n; i++) cin >> a[i]; // 读取宝石价值数组
        
        sort(a.begin(), a.end()); // 将宝石价值按从小到大排序
        
        // 计算前缀和数组
        for(int i = 1; i <= n; i++) {
            sum[i] = sum[i - 1] + a[i - 1];
        }
        
        ll ans = 0; // 用于存放最大价值和
        
        // 遍历可能的删除组合
        for(int m = 0; m <= k; m++) {
            if(2 * m <= n && (k - m) <= n - 2 * m) {
                ll current_sum = sum[n - (k - m)] - sum[2 * m];
                ans = max(ans, current_sum);//表示删除这些宝石后的总和。
            }
        }
        
        cout << ans << "\n"; // 输出结果
    }
    return 0;
}

详细过程解释

  1. 输入读取

    • 读取测试用例的数量 t
  2. 处理每个测试用例

    • 读取宝石的数量 n 和处理次数 k
    • 初始化一个大小为 n 的数组 a,用于存储每个宝石的价值。
    • 初始化一个大小为 n + 1 的数组 sum,用于存储前缀和,初始值为 0。
    • 读取宝石价值,存入数组 a
    • 对数组 a 进行排序,以便后续操作方便。
    • 计算前缀和 sumsum[i] 表示数组 a 中前 i 个元素的和。
  3. 遍历可能的删除组合

    • 遍历 m 从 0 到 k,表示删除 2m 个最小宝石和 (k - m) 个最大宝石。
    • 对于每个 m,计算删除后的剩余元素的和:
      • sum[n - (k - m)] 表示删除 k - m 个最大宝石后,剩余元素的和。
      • sum[2 * m] 表示删除 2m 个最小宝石后的和。
      • current_sum = sum[n - (k - m)] - sum[2 * m] 表示删除这些宝石后的总和。
    • 更新 ans 为当前情况下的最大和。
  4. 输出结果

    • 在每个测试用例处理完成后,输出 ans

这样,通过逐步删除最小宝石和最大宝石,并计算剩余元素的和,我们可以找到在规定次数内最大化宝石总价值的方法。

相关推荐
机器学习之心6 分钟前
异常检测 | 高斯分布拟合算法异常数据检测(Matlab)
算法·数学建模·matlab·异常数据检测
Mr.1311 分钟前
什么是 C++ 中的多继承?它有哪些优缺点?什么是虚继承?为什么要使用虚继承?
c++
cdut_suye12 分钟前
C++11新特性探索:Lambda表达式与函数包装器的实用指南
开发语言·数据库·c++·人工智能·python·机器学习·华为
K.L.Zous23 分钟前
Arduino键盘
c++
fpcc35 分钟前
c++应用网络编程之十五Nagle算法
网络·c++
Ning_.37 分钟前
力扣第 66 题 “加一”
算法·leetcode
kitesxian38 分钟前
Leetcode437. 路径总和 III(HOT100)
算法·深度优先
YSRM40 分钟前
异或-java-leetcode
java·算法·leetcode
木向44 分钟前
leetcode:222完全二叉树的节点个数
算法·leetcode
Ning_.1 小时前
力扣第 67 题 “二进制求和”
算法·leetcode