C - 开餐馆
题目
蒜头君想开家餐馆. 现在共有 n 个地点可供选择。蒜头君打算从中选择合适的位置开设一些餐馆。这 n 个地点排列在同一条直线上。我们用一个整数序列 m1,m2,...mn 来表示他们的相对位置。由于地段关系, 开餐馆的利润会有所不同。我们用 pi 表示在 mi 处开餐馆的利润。
为了避免自己的餐馆的内部竞争,餐馆之间的距离必须大于 k。
请你帮助蒜头君选择一个总利润最大的方案。
输入格式
标准的输入包含若干组测试数据。
输入第一行是整数 T(1≤T≤1000),表明有 T 组测试数据。紧接着有 T 组连续的测试。每组测试数据有 33 行;
第 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 且为整数)。
输出格式
对于每组测试数据可能的最大利润。
Sample Input
2 3 11 1 2 15 10 2 30 3 16 1 2 15 10 2 30
Sample Output
40 30
思路
因为数据量不大,直接用dp记录到每个店的满足条件的最优解即可,复杂度O(n²)
代码
#include<bits/stdc++.h>
using namespace std;
int t, n, k, m[105], dp[105], maxn;
int main()
{
cin >> t;
while (t--)
{
//不重置数组的话答案可能会被上一次的结果影响到
memset(m, 0, sizeof(m)), memset(dp, 0, sizeof(dp));
cin >> n >> k;
for (int i = 1; i <= n; ++i)
{
cin >> m[i];
}
for (int i = 1; i <= n; ++i)
{
cin >> dp[i];//每个店的最低利润当然是自家店的利润
}
for (int i = 1; i <= n; ++i)
{
maxn = 0;
for (int j = 1; j <= i; ++j)//找从第1家店到第i家店的最优解
{
if (m[i] - m[j] > k)//如果距离满足两店之间大于k的条件
{
maxn = max(maxn, dp[j]);
}
}
dp[i] += maxn;//将最优解存入
}
sort(dp + 1, dp + n + 1);//整理最优解数组,找出其中最大值
cout << dp[n] << endl;
}
return 0;
}