1133: [POI2009]Kon
Time Limit: 10 Sec Memory Limit: 162 MBSubmit: 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
*/