牛客算法题：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的甜蜜值。

思路

先假设n个巧克力都在晚上吃，然后再选k个巧克力改完早上吃。这k个改为早上吃的巧克力一定要是  n个巧克力中早上甜蜜之减晚上甜蜜之最大的k个。

代码

#include<iostream>
#include<algorithm>
using namespace std;
int main()
{
    int n,k;
    cin>>n>>k;
    long long ans=0;
    int *a=new int[n];
    int *b=new int[n];
    for(int i=0;i<n;i++)
        cin>>a[i];
    for(int i=0;i<n;i++)
    {
        cin>>b[i];
        ans+=b[i];//先假设n个都是在晚上吃
        a[i]-=b[i];//如果第i个改为在早上吃，就能多收获a[i]-b[i]的甜蜜值
    }
    sort(a,a+n);
    for(int i=0;i<k;i++)
    {
        ans+=a[n-1-i];
    }
    cout<<ans;
    return 0;
}