【openjudge】开餐馆

描述

北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有n 个地点可供选择。小明打算从中选择合适的位置开设一些餐馆。这 n 个地点排列在同一条直线上。我们用一个整数序列m₁, m₂, ... 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 个地点的位置m1 , m2, ... mn ( 1000000 > mi > 0 且为整数,升序排列)
第3行:n 个地点的餐馆利润p1 , p2, ... pn ( 1000 > pi > 0 且为整数)输出对于每组测试数据可能的最大利润样例输入

23 111 2 1510 2 303 161 2 1510 2 30

样例输出

4030

【代码】

#include
#include
#include
using namespace std;
int T,n,k;
int a[105],v[105],pre[105],f[105];
int main(){
	scanf("%d",&T);
	for (int t=1;t<=T;++t){
		scanf("%d%d",&n,&k);
		memset(a,0,sizeof(a));
		memset(f,0,sizeof(f));
		memset(v,0,sizeof(v));
		memset(pre,0,sizeof(pre));
		for (int i=1;i<=n;++i)
		  scanf("%d",&a[i]);
		for (int i=1;i<=n;++i)
		  scanf("%d",&v[i]);
		for (int i=2;i<=n;++i)
		  for (int j=i-1;j>=1;--j)
		    if (a[i]-a[j]>k){
			  pre[i]=j; break;
			}
		f[1]=v[1];
		for (int i=2;i<=n;++i)
		  f[i]=max(f[i-1],f[pre[i]]+v[i]);
		printf("%d\n",f[n]);
	}
}