题目描述
小W发明了一个游戏,他在黑板上写出了一行数字a1,a2,a3,……,an,然后给你M个回合的机会,每会回你可以从中选择一个数字擦去它,接着剩下来的每个数字ai都要递减一个值bi。如此重复m个回合,所有你擦去的数字之和就是你所得的分数。
小W和他的好朋友小Y玩了这个游戏,可是他发现,对于每个给出的a和b序列,小Y的得分总比他高,所以他就很不服气。于是他想让你帮他算算,对于每个a和b序列,可以得到的最大得分是多少。
Input
输入文件的第一行是一个整数n(1<=n<=2000),表示数字个数;第二行一个整数m(1<=m<=n),表示回合数,接下来一行有n个不超过10000的正整数,a1,a2,a3,……,an表示原始序列,最后一行有n个不超过500的正整数,b1,b2,b3,……,bn,表示每回合每个数字递减的值。
Output
输出文件只有一个整数,表示最大的可能得分
分析
这题很明显是线性DP
首先DP前应该排下序
把分数高的排前面
然后状态转移方程
max(f[i-1][j],f[i-1][j-1]+a[i]-b[i]*(j-1))
#include <iostream>
#include <cstdio>
#include <memory.h>
using namespace std;
int n, m, a[2001], b[2001], f[2001][2001], i, j;
void qs(int l, int h) {
int i = l, j = h, mid = b[(l + h) / 2], t;
do {
while(b[i] > mid) i++;
while(b[j] < mid) j--;
if(i <= j) {
t = a[i];
a[i] = a[j];
a[j] = t;
t = b[i];
b[i] = b[j];
b[j] = t;
i++;
j--;
}
} while(i <= j);
if(i < h) qs(i, h);
if(l < j) qs(l, j);
}
int main() {
scanf("%d%d", &n, &m);
for(i = 1; i <= n; i++)
scanf("%d", &a[i]);
for(i = 1; i <= n; i++)
scanf("%d", &b[i]);
qs(1, n);
memset(f, 200, sizeof(f));
for(i = 1; i <= n; i++)
for(j = 1; j <= m; j++) {
f[i - 1][0] = 0;
f[i][j] = max(f[i - 1][j], f[i - 1][j - 1] + a[i] - b[i] * (j - 1));
}
printf("%d", f[n][m]);
}