POJ 4118 开餐馆

题目

总时间限制: 1000ms 内存限制: 65536kB
描述
北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有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[i]=max(dp[i], dp[j]+p[i]),j=1,2,3,...,i-1 \ and \ (m[i]-m[j])>k$

代码

while True:
    try:
        T = int(input().strip())
        for i in range(T):
            n, k = input().strip().split()
            n = int(n); k = int(k)
            m = [int(i) for i in input().strip().split()]
            p = [int(i) for i in input().strip().split()]
            dp = [p[i] for i in range(n)]
            for i in range(n):
                for j in range(i):
                    if (m[i] - m[j] > k):
                        dp[i] = max(dp[i], dp[j] + p[i])
            print(max(dp))
    except:
        break