总时间限制: 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
解题思路
dp[i]为在前i个地点中选择地点开餐馆且确定要在第i地点开餐馆,餐馆之间距离大于k,能够获得的最大利润。
第i地点开餐馆的利润为p[i],第i地点的位置为m[i]。
如果只在第i地点开餐馆,其它地方不开,那么能获得的最大利润为dp[i] = p[i]
如果在第i地点餐馆的前面的位置还要开餐馆,那么前一个餐馆的位置为第j地点。由于两餐馆间距离必须大于k,所以必须满足m[i] - m[j] > k。
j从1开始循环到i-1,同时要满足m[i] - m[j] > k,对于每一个可能的第i地点餐馆前一个餐馆的地点j,可以得到:前j个地点中选择地点开餐馆,且在第j地点开餐馆能获得的最大利润dp[j],再加上第i地点开餐馆的利润p[i],即为在前i个地点中选择地点开餐馆且在第i地点开餐馆能得到的最大利润。dp[i] = dp[j] + p[i]
以上所有情况取最大值
#include<bits/stdc++.h>
using namespace std;
#define N 105
int dp[N], m[N], p[N];//dp[i]:在前i个地点中选择地点开餐馆且确定要在第i位置开餐馆,餐馆之间距离大于k,能够获得的最大利润。
int main()
{
int t, n, k, mx;
cin >> t;
while(t--)
{
memset(dp, 0, sizeof(dp));//注意数据清空
mx = 0;//最大利润
cin >> n >> k;
for(int i = 1; i <= n; ++i)
cin >> m[i];
for(int i = 1; i <= n; ++i)
cin >> p[i];
for(int i = 1; i <= n; ++i)
{
dp[i] = p[i];
for(int j = 1; j < i && m[i] - m[j] > k; ++j)
dp[i] = max(dp[i], dp[j] + p[i]);
mx = max(mx, dp[i]);
}
cout << mx << endl;
}
return 0;
}