百练 开餐馆

百练 开餐馆

总时间限制: 内存限制:
1000ms 65536kB

描述

北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有n 个地点可供选择。小明打算从中选择合适的位置开设一些餐馆。这 n 个地点排列在同一条直线上。我们用一个整数序列$m_1, m_2, ... m_n$ 来表示他们的相对位置。由于地段关系,开餐馆的利润会有所不同。我们用$p_i$ 表示在$m_i$ 处开餐馆的利润。为了避免自己的餐馆的内部竞争,餐馆之间的距离必须大于k。请你帮助小明选择一个总利润最大的方案。

输入

标准的输入包含若干组测试数据。输入第一行是整数T (1 <= T <= 1000) ，表明有T组测试数据。紧接着有T组连续的测试。每组测试数据有3行,

• 第1行:地点总数 n (n < 100), 距离限制 k (k > 0 && k < 1000).
• 第2行:n 个地点的位置$m_1 , m_2, ... m_n$ ( 1000000 > $m_i$ > 0 且为整数,升序排列)
• 第3行:n 个地点的餐馆利润$p_1 , p_2, ... p_n$ ( 1000 > $p_i$ > 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;
}