当前位置: 代码迷 >> 综合 >> 百炼 6045 开餐馆问题
  详细解决方案

百炼 6045 开餐馆问题

热度:96   发布时间:2023-09-22 07:05:23.0
题目如下:描述

北大信息学院的同学小明毕业之后打算创业开餐馆.现在共有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 且为整数)输出对于每组测试数据可能的最大利润样例输入
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;elsereturn 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];//即为0elseprice = a[q - k - 1];a[q] = max(a[q - 1], price + node[search(q)].p);}}printf("%d", a[len]);return 0;
}