OpenJudge 4118: 开餐馆

OpenJudge 4118: 开餐馆

北大2017大数据研究中心夏令营上机考试 E题
北大百炼OJ/练习 全局题号：6045
总时间限制:1000ms，内存限制:65536kB
原题OJ链接

描述

北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有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），代码中核心的式子为：dp[x]=max(dp[x],dp[y]+p[x])，其中dp[x]表示前x个地点中开餐馆的最大盈利。dp数组的初始化值为输入中的每个餐馆的盈利p1, p2, … pn。

源代码

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int main(){
    int T,n,k; //T, n, k分别于题目中的输入对应
    int m[200],p[200],dp[200];
    cin>>T;
    for(int i=0;i<T;i++){
        cin>>n>>k;
        for(int j=0;j<n;j++){
            cin>>m[j]; // m[j]记录餐馆地点位置
        }
        for(int j=0;j<n;j++){
            cin>>p[j]; // p[j]记录每个餐馆的盈利
            dp[j]=p[j]; // dp[j]初始化
        }

        for(int x=0;x<n;x++){
            for(int y=0;y<x;y++){
                if((m[x]-m[y])>k){
                    dp[x]=max(dp[x],dp[y]+p[x]);
                    // dp[x]记录前x个地点开餐馆的最大盈利
                }
            }
        }

        int profit=0;
        for(int j=0;j<n;j++){
            if(dp[j]>profit) profit=dp[j];
        }
        cout<<profit<<endl;

    }    
    return 0;    
}