百炼 6045 开餐馆问题

题目如下：描述

北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有n 个地点可供选择。小明打算从中选择合适的位置开设一些餐馆。这 n 个地点排列在同一条直线上。我们用一个整数序列m₁, m₂, ... m_n 来表示他们的相对位置。由于地段关系,开餐馆的利润会有所不同。我们用p_i 表示在m_i 处开餐馆的利润。为了避免自己的餐馆的内部竞争,餐馆之间的距离必须大于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 且为整数)输出对于每组测试数据可能的最大利润样例输入

23 111 2 1510 2 303 161 2 1510 2 30

样例输出

4030

该题目类似于背包，用了一种动态规划的思想。

此题一开始犯了一个错，search返回下标包含0的情况与a[0]重叠了，后来从1开始存就没问题了。代码如下（如有错误或者不够简练的地方，请各位大佬批评指正）：

// restuarant.cpp: 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <stdio.h>
#include <stdlib.h>
#include <algorithm>
#pragma warning(disable:4996)
struct node
{
	int m;
	int p;
}node[100];
int a[1000000];
int n, k;
int search(int s)//对应位置有没有点，如果有的话返回对应饭馆的下标，没有则返回0
{
	for (int j = 1; j < n+1; j++)
	{
		if (node[j].m == s)
		{
			return j;
		}
	}
	return 0;
}

int max(int i, int j)
{
	if (i < j)
		return j;
	else
		return i;
}

int main()
{
	
	scanf("%d", &n);
	scanf("%d", &k);
	for (int i = 1; i < n+1; i++)
	{
		scanf("%d", &node[i].m);
	}
	for (int i = 1; i < n+1; i++)
		scanf("%d", &node[i].p);
	//for (int j = 0; j < n; j++)
	//	printf("%d %d\n", node[j].m, node[j].p);
	a[0] = 0;
	int len = node[n].m;
	for (int q = 1; q <= len; q++)
	{
		if (search(q) == 0)
			a[q] = a[q - 1];
		else
		{
			int price;
			if (q < k)
				price = a[0];//即为0
			else
				price = a[q - k - 1];
			a[q] = max(a[q - 1], price + node[search(q)].p);
		}
	}
	printf("%d", a[len]);
    return 0;
}