百练 开餐馆
总时间限制: 内存限制:
1000ms 65536kB
描述
北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有n 个地点可供选择。小明打算从中选择合适的位置开设一些餐馆。这 n 个地点排列在同一条直线上。我们用一个整数序列 m1,m2,...mn 来表示他们的相对位置。由于地段关系,开餐馆的利润会有所不同。我们用 pi 表示在 mi 处开餐馆的利润。为了避免自己的餐馆的内部竞争,餐馆之间的距离必须大于k。请你帮助小明选择一个总利润最大的方案。
输入
标准的输入包含若干组测试数据。输入第一行是整数T (1 <= T <= 1000) ,表明有T组测试数据。紧接着有T组连续的测试。每组测试数据有3行,
- 第1行:地点总数 n (n < 100), 距离限制 k (k > 0 && k < 1000).
- 第2行:n 个地点的位置 m1,m2,...mn ( 1000000 > mi > 0 且为整数,升序排列)
- 第3行:n 个地点的餐馆利润 p1,p2,...pn ( 1000 > pi > 0 且为整数)
输出
对于每组测试数据可能的最大利润
样例输入
2
3 11
1 2 15
10 2 30
3 16
1 2 15
10 2 30
样例输出
40
30
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAX_NUM = 1000 + 5;
int n;
int dis[MAX_NUM];
int pro[MAX_NUM];
int k;
int ans;
int maxP[MAX_NUM];
void dfs(int id, int p) {
if(dis[id] + k >= dis[n - 1]) {
ans = max(p, ans);
return;
}
for(int i = id + 1; i < n; i++) {
// 最优性剪枝
// 到当前位置的最大值已不大于记录的到当前位置的值
if(dis[i] - dis[id] > k && p + pro[i] > maxP[i]) {
maxP[i] = p + pro[i];
dfs(i, p + pro[i]);
}
}
}
int main() {
int kase;
scanf("%d", &kase);
while(kase--) {
memset(dis, 0, sizeof(dis));
memset(pro, 0, sizeof(pro));
memset(maxP, 0, sizeof(maxP));
ans = 0;
scanf("%d%d", &n, &k);
for(int i = 0; i < n; i++) {
scanf("%d", &dis[i]);
}
for(int i = 0; i < n; i++) {
scanf("%d", &pro[i]);
}
for(int i = 0; i < n; i++) {
dfs(i, pro[i]);
}
printf("%d\n", ans);
}
return 0;
}