GDUT - 专题学习2 C - 开餐馆

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;
}