牛客题解记录–装进肚子
装进肚子
题目描述
自从ZZZZone吃完糖果后,他开始改吃巧克力了,他每天想吃n个巧克力增在甜蜜值,他决定早上吃K个巧克力,晚上吃n - K个巧克力,每个巧克力在早上吃和在晚上吃的甜蜜值是不一样的,他想让自己得到的甜蜜值最大,并想知道最大是多少。
请你编程帮助他。
输入描述:
第一行包含两个数n,K表示每天要吃的巧克力数量和要在早上吃的数量。(n <= 100000, K <= n)
第二行包含n个整数Ai(1 <= i <= n) 表示个第i个巧克力在早上吃可得到的甜蜜值 (Ai <= 100000)
第三行包含n个整数Bi(1 <= i <= n) 表示个第i个巧克力在晚上吃可得到的甜蜜值 (Bi <= 100000)
输出描述:
输出仅一行包含一个整数表示ZZZZone能获得的最大甜蜜值。
示例
输入:
2 1
3 6
2 8
输出:
11
贪心思路
ACM类型的贪心类型题目,有一个共性就是一定会将具体的题目给你具象化,将抽象数据具象为自然语言,增加理解题目的难度。
这个题目想让你去计算白天和晚上吃巧克力能获得的最大甜蜜度,乍一看感觉需要去动态规划求最大,但是如果你将这个题抽象为在同index(白天a[index]和晚上b[index]传入数据的次序)下的两个值我需要拿取策略最优的那就简单多了,我们只需要将整个数据根据index绑定,并且根据 ( a 1 − b 1 ) − ( a 2 − b 2 ) (a_{1}- b_{1}) - (a_{2}- b_{2}) (a1−b1)−(a2−b2)来进行排序,获取一个根据白天与晚上差值小到大的有序数据,这样我们就能获取最优的白天或者晚上的吃巧克力的策略。
代码实现
import java.util.*;
import java.io.*;
public class Main {
static BufferedReader ins = new BufferedReader(new InputStreamReader(System.in));
static StreamTokenizer in = new StreamTokenizer(ins);
static PrintWriter out = new PrintWriter(new OutputStreamWriter(System.out));
static int[][] nums = new int[100002][3];//定义一个空间为测试数据的边界长度的数组
public static void main(String[] args) throws IOException {
in.nextToken();
int n = (int) in.nval; // 读取输入的n值,表示数组长度
in.nextToken();
int sun = (int) in.nval; // 读取输入的sun值,白天吃巧克力数量
int nit = n - sun; // 根据白天吃巧克力数量计算晚上数量
// 初始化nums数组
for (int i = 0; i < 100002; i++) {
Arrays.fill(nums[i], 0);
}
// 读取输入的数组元素的值,白天吃巧克力
for (int i = 0; i < n; i++) {
in.nextToken();
nums[i][0] = (int) in.nval;
}
// 读取输入的数组元素的值,晚上吃巧克力,并计算差值
for (int i = 0; i < n; i++) {
in.nextToken();
nums[i][1] = (int) in.nval;
nums[i][2] = nums[i][1] - nums[i][0];
}
ins.close();
// 根据差值对nums数组进行排序
Arrays.sort(nums, 0, n, new Comparator<int[]>() {
public int compare(int[] a, int[] b) {
return a[2] - b[2];
}
});
long ans = 0;
// 计算白天吃巧克力的和
for (int i = 0; i < sun; i++) {
ans += nums[i][0];
}
// 计算晚上吃巧克力的和
for (int i = sun; i < n; i++) {
ans += nums[i][1];
}
out.print(ans);
out.close();
}
}