dp。注意值是非负数。。所以会出现所有数据都是0的情况,所以dp初值要赋成-1。这道题的思想挺巧妙的,首先我们分析一下题目,其实可以发现,如果边 i -i+1 被选中,那么i点以前的点对i+1点以后的所以点就都没有影响了。因为如果存在边c[ i ][ j ] > 0 那么肯定在边i - i+1时就已经happy了。所以对于以后的情况就不用考虑了。这样的话就可以把原题分解成一些子区间。所以我们可以用g[ i ][ j ]表示点i 到点 j只有 (j-1) - j 这条边被选中了。于是就可以想到了转移状态:
dp[ j ][ t ] = max(dp[ i ][ t - 1 ] + g[ i ][ j ])。
dp[ j ][ t ]表示从点1到 点j选中t条边happy的最大人数。然后就可以用三层循环递推出所有dp的值了。
#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
#include<cstdio>
#include<cmath>
#define CLR(a, b) memset(a, b, sizeof(a))
#define REP(i, n) for(int i=0; i<n; i++)
#define FF(i, a, b) for(int i=a; i<b; i++)
#define FD(i, a, b) for(int i=a; i>=b; i--)
#define LL long long
#define PB push_back
using namespace std;
const int N = 555;
const int INF = 1e9;
int dp[N][N], pt[N][N];
int c[N][N], g[N][N];
int n, k;
void pts(int r, int st)
{
if(st == 0) return;
pts(pt[r][st], st - 1);
printf("%d", r - 1);
if(st == k) puts("");
else putchar(' ');
}
int main()
{
while(scanf("%d%d", &n, &k) != EOF)
{
CLR(c, 0);
for(int i = 1; i < n; i ++)
for(int j = i + 1; j <= n; j ++)
{
scanf("%d", &c[i][j]);
c[i][j] += c[i][j - 1];
}
for(int i = 1; i < n; i ++)
for(int j = i + 1; j <= n; j ++)
{
g[i][j] = 0;
for(int t = i; t < j; t ++)
{
g[i][j] += c[t][n] - c[t][j - 1];
}
}
CLR(dp, -1);
dp[1][0] = 0;
for(int i = 1; i <= n; i ++)
for(int j = i + 1; j <= n; j ++)
for(int t = 1; t <= min(i, k); t ++)
{
if(dp[j][t] < dp[i][t - 1] + g[i][j])
dp[j][t] = dp[i][t - 1] + g[i][j], pt[j][t] = i;
}
int ans = -1, id;
for(int i = k + 1; i <= n; i ++)
{
if(dp[i][k] > ans)
{
ans = dp[i][k];
id = i;
}
}
printf("%d\n", ans);
pts(id, k);
}
return 0;
}
/*
5 3
5 0 6 5
5 3 7
5 3
10
4 2
5 0 6
5 3
5
4 3
5 0 6
5 3
5
*/