T95要减肥(动态规划)

链接:
https://www.nowcoder.com/acm/contest/79/B

来源:牛客网

时间限制:C/C++ 1秒,其他语言2秒

空间限制:C/C++ 262144K,其他语言524288K

64bit IO Format: %lld

题目描述

T95提速啦,然而她觉得自己还是太慢了!

于是T95决定减肥——然而T95还喜欢吃麦当劳

现在有n种赛道,n家麦当劳,第i条赛道的痛苦值是ai,第i家麦当劳的快乐值是bi

为了减肥,T95吃的麦当劳次数不能大于她跑步的次数

由于T95太重了,每个赛道跑一次之后就会坏掉

由于T95喜欢吃不同的口味,每家麦当劳最多吃一次

由于明斯克航空航天局的蜜汁bug,T95每跑3次就会得到额外的m块乐值

现在T95想知道她可以得到的最大(快乐值的和-痛苦值的和)是多少

输入描述:

第一行两个数n,m

第二行n个数表示ai

第三行n个数表示bi

输出描述:

输出一行一个数表示答案
代码:
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
long long a[1000010];
long long b[1000010];
long long f[1000010];
long long max1(long long x,long long y)
{
    if(x>=y)
        return x;
    return y;
}
int  main()
{
    long long n,m;
    scanf("%lld%lld",&n,&m);
    long long i;
    memset(f,0,sizeof(f));
    for(i=1; i<=n; i++)
    {
        scanf("%lld",&a[i]);
    }
    sort(a+1,a+n+1);
    for(i=1; i<=n; i++)
    {
       scanf("%lld",&b[i]);
    }
    sort(b+1,b+n+1);
    long long k=0;
    for(i=1; i<=n; i++)
    {
        f[i]=f[i-1]+b[n-i+1]-a[i];
        if(i%3==0)
        {
            if(f[i-3]<(f[i]+m))
            {
                f[i]=f[i]+m;
            }
            else
            {
                 f[i]=f[i-3];
                 k=i;
                 break;
            }
            k=i;
        }
    }
    f[n]=max1(f[n],f[k]);
    printf("%lld\n",f[n]);
    return 0;
}

点赞