计蒜客T1224 开餐馆

题目

蒜头君想开家餐馆. 现在共有 nn 个地点可供选择。蒜头君打算从中选择合适的位置开设一些餐馆。这 nn 个地点排列在同一条直线上。我们用一个整数序列 m_1, m_2, ... m_nm1​,m2​,...mn​ 来表示他们的相对位置。由于地段关系, 开餐馆的利润会有所不同。我们用 p_ipi​ 表示在 m_imi​ 处开餐馆的利润。

为了避免自己的餐馆的内部竞争,餐馆之间的距离必须大于 kk。

请你帮助蒜头君选择一个总利润最大的方案。

输入格式

标准的输入包含若干组测试数据。

输入第一行是整数 T (1 \le T \le 1000)T(1≤T≤1000)，表明有 TT 组测试数据。紧接着有 TT 组连续的测试。每组测试数据有 33 行;

第 11 行:地点总数 n (n < 100)n(n<100), 距离限制 k (k > 0k(k>0 && k < 1000)k<1000)；

第 22 行: nn 个地点的位置 m_1 , m_2, ... m_n ( 1000000 > m_i > 0m1​,m2​,...mn​(1000000>mi​>0 且为整数, 升序排列)；

第 33 行: nn 个地点的餐馆利润 p_1 , p_2, ... p_n ( 1000 > p_i > 0p1​,p2​,...pn​(1000>pi​>0 且为整数)。

输出格式

对于每组测试数据可能的最大利润。

思路

动态规划。对于每一家餐馆都有选或不选两种选择，可以推出递归式：

dp[i][j] = max(p[i]+dp[i+1][m[i]+k], dp[i+1][j])

各字母的含义与题目中相同，注意限制条件

代码

#include<iostream>
#include<stdio.h>


int dp[105];

int main()
{
	int T;
	scanf("%d", &T);
	
	int n, k;
	int m[105];
	int p[105];
	
	int j;
	int case1, case2;
	while(T--)
	{
		scanf("%d %d", &n, &k);
		for(int i = 1; i <= n; i++) scanf("%d", &m[i]);
		for(int i = 1; i <= n; i++) scanf("%d", &p[i]);
		
		dp[n] = p[n];
		for(int i = n-1; i >= 1; i--)
		{
			//printf("judge NO.%d\n", i);
			j = i+1;
			while(j <= n && m[j] <= m[i]+k) j++;
			if(j > n) case1 = p[i];
			else case1 = p[i] + dp[j];
			
			case2 = dp[i+1];
			
			//printf("case1=%d\ncase2=%d\n", case1, case2);
			dp[i] = case1 > case2 ? case1 : case2;
			//printf("dp[%d]=%d\n", i, dp[i]);
		}	
		
		printf("%d\n", dp[1]);
	}
	
	
	return 0;
}