牛客竞赛算法入门题单打卡 B装进肚子

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

自从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能获得的最大甜蜜值。

示例1

输入

复制2 1 3 6 2 8

2 1
3 6
2 8

输出

复制11

11

说明

早上吃第一个巧克力得到3甜蜜值，晚上吃第2个巧克力得到8的甜蜜值，所以最大可得到11的甜蜜值。

思路

贪心算法，先假设全部都是晚上吃的，然后将早上与晚上的甜蜜值差值放到一个列表里降序排序，将该列表里的前k个差值加到答案中即可得到最终答案。

易错点：

这里不是将每种食物甜蜜值取早晚中大的那个加起来，而是要在规定条件下找到一个最优方案，使甜蜜值最大

#include <bits/stdc++.h>
#include <algorithm>
using namespace std;

bool cmp(int x, int y) {
	return x > y;
}

int main() {
	int n, k;
	cin >> n >> k;
	long long ans = 0;
	int a[n], b[n], c[n];
	for (int i = 0; i < n; i++)
		cin >> a[i];
	for (int i = 0; i < n; i++) {
		cin >> b[i];
		c[i] = a[i] - b[i];
		ans += b[i];
	}
	sort(c, c + n, cmp);
	long long arr = 0;
	for (int i = 0; i < k; i++) {
		arr += c[i];
	}
	cout << arr + ans;
}