bzoj 1133: [POI2009]Kon（DP）

1133: [POI2009]Kon

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 457   Solved: 166
[ Submit][ Status][ Discuss]

Description

火车沿途有N个车站，告诉你从每一站到每一站的人数，现在查票员只能查K次票，每次查票可以控制目前在车上的所有乘客的车票。求一个查票方案，使得控制的不同的乘客尽量多。 （显然对同一个乘客查票多次是没有意义的，只算一次）

Input

第一行正整数 N K (1≤K＜N≤600， K≤50). 接下来N-1行，第i行第j个数描述第i站上，到第i+j站下的乘客个数。总乘客数≤2*10^9

Output

单调增的K个整数，用空格隔开，表示经过哪些站以后查票。

Sample Input

7 2

2 1 8 2 1 0

3 5 1 0 1

3 1 2 2

3 5 6

3 2

1

Sample Output

2 5

dp[k][x]表示前x站检查k次票，且最后一次检票正好是在第x站的最大值

有dp[k][x] = max(dp[k-1][i]+sum[i+1][x]  (k-1<=i<x) )

其中sum[x][y]表示出发点在x之后，终点在y之后的乘客总数量

sum[][]可以n^3预处理，然后n²k的dp就ok了

记录路径

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef struct
{
	int val;
	int last;
}Res;
Res dp[52][605];
int a[605][605], sum[605][605];
void Print(int k, int id)
{
	if(k==0)
		return;
	Print(k-1, dp[k][id].last);
	if(k!=1)
		printf(" ");
	printf("%d", id);
}
int main(void)
{
	int n, k, i, j, p, ans, temp;
	scanf("%d%d", &n, &k);
	n -= 1;
	for(i=1;i<=n;i++)
	{
		for(j=i;j<=n;j++)
			scanf("%d", &a[i][j]);
	}
	for(i=n;i>=1;i--)
	{
		for(j=i+1;j<=n;j++)
			sum[i][j] = sum[i+1][j];
		for(j=i;j<=n;j++)
		{
			for(p=i;p<=j;p++)
				sum[i][p] += a[i][j];
		}
	}
	memset(dp, -1, sizeof(dp));
	dp[0][0].val = 0;
	for(i=1;i<=k;i++)
	{
		for(j=i;j<=n;j++)
		{
			for(p=i-1;p<=j-1;p++)
			{
				if(dp[i-1][p].val!=-1 && dp[i-1][p].val+sum[p+1][j]>dp[i][j].val)
					dp[i][j].val = dp[i-1][p].val+sum[p+1][j], dp[i][j].last = p;
			}
		}
	}
	ans = -1;
	for(i=k;i<=n;i++)
	{
		if(dp[k][i].val>ans)
			ans = dp[k][i].val, temp = i;
	}
	Print(k, temp);
	printf("\n");
	return 0;
}
/*
15 6
1 9 8 7 2 0 9 8 0 9 9 8 0 0
1 2 3 9 8 7 6 5 1 4 2 3 5
1 9 8 7 2 3 6 5 4 2 1 6
8 7 2 3 4 7 6 5 1 5 6
1 8 7 6 2 3 5 1 5 6
9 8 7 6 5 4 3 2 1
1 2 3 4 5 6 7 8
7 6 5 4 3 2 1
1 2 3 4 5 6
5 4 3 2 1
1 2 3 4
3 2 1
1 2
1

5 4
0 0 0 0
0 0 0
0 0
0
*/