dp[i][j]:第j次检查选第i个车站检查的最多人数
f[i][j]:表示i<=起点<=j && 终点>j的的人数
t[i][j]:表示起点为i,终点>j的人数
ans[i][j]:dp[i][j]最优时,前面一个转移来的车站是什么
注意:题目要求输出递增,说明必须选满k个,所以dp[i][j]的i>=j,不然有可能会选不满k个,最后一个点会wa
#include <bits/stdc++.h>
using namespace std;
#define N 700
int n,m,a[N][N],t[N][N],f[N][N],dp[N][N],ans[N][N],mx,s;
void print(int x,int k){
if(k==0) return;
print(ans[x][k],k-1);
printf("%d ",x);
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<n;i++)
for(int j=i+1;j<=n;j++)
scanf("%d",&a[i][j]);
for(int i=1;i<n;i++)
for(int j=n-1;j>=i;j--)
t[i][j]=t[i][j+1]+a[i][j+1];
for(int i=n-1;i>=1;i--)
for(int j=i;j<n;j++)
f[i][j]+=f[i+1][j]+t[i][j];
for(int k=1;k<=m;k++)
for(int i=k;i<n;i++)
for(int j=k-1;j<i;j++)
if(dp[i][k]<dp[j][k-1]+f[j+1][i]){
dp[i][k]=dp[j][k-1]+f[j+1][i];
ans[i][k]=j;
}
for(int i=1;i<n;i++)
if(mx<dp[i][m]){
mx=dp[i][m];
s=i;
}
print(s,m);
return 0;
}