开餐馆

004:开餐馆

总时间限制: 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 <iostream>
#include <cstdio>
using namespace std;
int m[105],p[105],b[105];
int main()
{
    int t;
    cin>>t;
    while(t--){
        int n,k,maxSum=0;
        cin>>n>>k;
        for(int i=0;i<n;i++){
            cin>>m[i];
        }
        for(int i=0;i<n;i++){
            cin>>p[i];
        }
        if(n==1) cout<<p[0]<<endl;
        else{
            b[0]=p[0];
            int cnt=1;
            for(int i=1;m[i]-m[0]<=k;i++){
                b[i]=p[i];
                cnt++;
            }
            for(int j=cnt;j<n;j++){
                int max=0;
                for(int i=j-1;i>=0;i--){
                    if(m[j]-m[i]>k && b[i]>max){
                        max=b[i];
                        if(m[i]-m[i-1]>k) break;
                    }
                }
                b[j]=p[j]+max;
            }
            for(int i=n-1;m[n-1]-m[i]<=k;i--){
                if(b[i]>maxSum){
                    maxSum=b[i];
                }
            }

            cout<<maxSum<<endl;
            }
        }

    return 0;
}