Wannafly挑战赛12 B - T95要减肥 dp

链接： 点击打开链接
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

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

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

现在有n种赛道，n家麦当劳，第i条赛道的痛苦值是a_i，第i家麦当劳的快乐值是b_i

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

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

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

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

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

输入描述:

第一行两个数n,m

第二行n个数表示a_i

第三行n个数表示b_i

输出描述:

输出一行一个数表示答案

a从小到大，b从大到小分别排序， c[i] = b[i] - a[i]

状态转移公式 dp[i] = dp[i-1] + c[i] + (i + 1) % 3 == 0 ? m : 0;

#include
using namespace std;
typedef long long ll;
typedef pairP;
const int inf = 0x3f3f3f3f;
const int N = 1000000 + 5;
const int mod = 1e9 + 7;
ll a[N], b[N], c[N], dp[N];
bool cmp(const int&a, const int&b)
{
    return a > b;
}
int main()
{
    ll n, m;
    scanf("%lld%lld", &n, &m);
    for(int i = 0; i < n; i++) scanf("%lld", &a[i]);
    for(int i = 0; i < n; i++) scanf("%lld", &b[i]);
    sort(a, a + n); sort(b, b + n, cmp);
    ll ans = 0LL;
    for(int i = 0; i < n; i++)
    {
        c[i] = b[i] - a[i];
    }
    dp[0] = c[0];
    ll maxi = dp[0];
    for(int i = 1; i < n; i++)
    {
        dp[i] = c[i];
        if((i+1)%3 == 0)
            dp[i] += m;
        dp[i] += dp[i-1];
        maxi = max(maxi, dp[i]);
    }
    printf("%lld\n", maxi);
}