洛谷-P1622 释放囚犯

题目描述

Caima王国中有一个奇怪的监狱，这个监狱一共有P个牢房，这些牢房一字排开，第i个紧挨着第i+1个（最后一个除外）。现在正好牢房是满的。

上级下发了一个释放名单，要求每天释放名单上的一个人。这可把看守们吓得不轻，因为看守们知道，现在牢房中的P个人，可以相互之间传话。如果某个人离开了，那么原来和这个人能说上话的人，都会很气愤，导致他们那天会一直大吼大叫，搞得看守很头疼。如果给这些要发火的人吃上肉，他们就会安静点。
输入输出格式

输入格式：

第一行两个数P和Q，Q表示释放名单上的人数；

第二行Q个数，表示要释放哪些人。

【数据规模】

对于100%的数据1≤P≤1000； 1≤Q≤100；Q≤P；且50%的数据 1≤P≤100；1≤Q≤5

输出格式：

仅一行，表示最少要给多少人次送肉吃。
输入输出样例

输入样例#1：

    20 3
    3 6 14

输出样例#1：

35

【样例说明】

先释放14号监狱中的罪犯，要给1到13号监狱和15到20号监狱中的19人送肉吃；再释放6号监狱中的罪犯，要给1到5号监狱和7到13号监狱中的12人送肉吃；最后释放3号监狱中的罪犯，要给1到2号监狱和4到5号监狱中的4人送肉吃。

——————————————————————————————————————————————————————
---------------------
思路：

将所有要释放的囚犯按照从小到大的顺序排好。
dp[i][j]为最后释放第i个到第j个囚犯所需代价，这时a[i-1]和a[j+1]已经被释放。

假设先释放i和j之间的第k个囚犯，则释放第k个囚犯所需的肉是a[j+1] - a[i-1] - 1 - 1。

a[j+1] - a[i-1] -1是i和j之间的囚犯总数，最后一个减一是因为第k个囚犯不需要肉。

则：dp[i][j] = min(dp[i][k-1] + dp[k+1][j] + a[j+1] - a[i-1] - 1 - 1); (i <= k <= j)

#include <iostream>  
#include <algorithm>
#include <cstdio>
#include <cstring>  
using namespace std;  
  
const int MAXN = 10001;
const int INF = 0x3f3f3f3f;
int a[MAXN], dp[MAXN][MAXN];

int main()  
{  
	int p, q;
	scanf("%d%d", &p, &q);

	for (int i = 1; i <= q; i++)
	{
		scanf("%d", &a[i]);
	}

	a[0] = 0;
	a[q+1] = p + 1;
	sort(a+1, a+q+1);
	memset(dp, 0, sizeof(dp));

	for (int len = 0; len <= q; len++)
	{
		for (int i = 1; i + len <= q; i++)
		{
			int j = i + len;
			dp[i][j] = INF;
			for (int k = i; k <= j; k++)
			{
				dp[i][j] = min(dp[i][j], dp[i][k-1] + dp[k+1][j] + a[j+1] - a[i-1] - 1 - 1);
			}
		}
	}

	printf("%d\n", dp[1][q]);
    return 0;  
}