YUT 3145 动态规划进阶题目之开餐馆

版权声明：转载请附上原文链接哟！ https://blog.csdn.net/weixin_44170305/article/details/90056311

风华是一指流砂，苍老是一段年华。

题目描述

北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有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 且为整数) 

输出

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

样例输入

copy

2
3 11
1 2 15
10 2 30
3 16
1 2 15
10 2 30

样例输出

40
30

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
int a[1005],b[1005],dt[1005][2];
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        memset(dt,0,sizeof(0));
        int i,j,n,k,p;
        scanf("%d %d",&n,&k);
        for(i=1; i<=n; i++)
            scanf("%d",&a[i]);
        for(i=1; i<=n; i++)
            scanf("%d",&b[i]);
        dt[1][0]=0;
        for(i=1; i<=n; i++)
        {
            if(a[i]-k<a[1])
                dt[i][1]=b[i];
            else
            {
                for(j=i-1; j>=1; j--)
                {
                    if(a[i]-k>a[j])
                        break;
                }
                dt[i][1]=max(dt[j][0],dt[j][1])+b[i];
            }
            dt[i][0]=max(dt[i-1][1],dt[i-1][0]);
        }
        p=max(dt[n][0],dt[n][1]);
        cout<<p<<endl;
    }
    return 0;
}