描述
北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有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过程其实不难,最后的输出WA到我了。
- 每个餐厅都更新它之前的最优解,不要觉得推到最后 f[n] 就是答案
假如:
n=3 , k=10
pos[ ]:1 , 12 ,13
val [ ]:1 ,100 ,1
所以,n=3时 选最优解,就会跳过 ans
一看代码就懂啦:
#include<stdio.h> #include<algorithm> #include<string.h> using namespace std; int pos[101],val[101],f[101]; int main() { int T; scanf("%d",&T); while(T--) { int n,k; scanf("%d%d",&n,&k); for(int i=1;i<=n;++i) scanf("%d",&pos[i]); for(int i=1;i<=n;++i) scanf("%d",&val[i]); for(int i=1;i<=n;++i) { f[i]=val[i]; for(int j=1;j<i;++j) { if(pos[i]-pos[j]>k) { f[i]=max(f[i],f[j]+val[i]); } } } int ans=0; for(int i=1;i<=n;++i) ans=max(ans,f[i]); printf("%d ",ans); memset(f,0,sizeof(f)); } return 0; }