传送门
极其简单的dp题
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
void read(int &x) {
char ch; bool ok;
for(ok=0,ch=getchar(); !isdigit(ch); ch=getchar()) if(ch=='-') ok=1;
for(x=0; isdigit(ch); x=x*10+ch-'0',ch=getchar()); if(ok) x=-x;
}
#define rg register
const int maxn=3010;
int n,m,f[60][610][2],a[610][610],sum[610][610],ans[60];
int main()
{
read(n),read(m);
for(rg int i=1;i<=n;i++)for(rg int j=i+1;j<=n;j++)read(a[i][j]);
for(rg int i=1;i<=n;i++)
for(rg int j=n;j>=1;j--)
sum[i][j]=sum[i-1][j]+sum[i][j+1]+a[i][j]-sum[i-1][j+1];
for(rg int i=1;i<=m;i++)
for(rg int j=i;j<=n;j++)
for(rg int k=i-1;k<j;k++)
if(sum[j][j+1]-sum[k][j+1]+f[i-1][k][0]>f[i][j][0])
f[i][j][0]=sum[j][j+1]-sum[k][j+1]+f[i-1][k][0],f[i][j][1]=k;
for(rg int i=m;i<=n;i++)if(f[m][i][0]>f[m][ans[m]][0])ans[m]=i;
for(int i=m,j=ans[m];i>=1;j=f[i][j][1],i--)ans[i]=j;
for(rg int i=1;i<=m;i++)printf("%d ",ans[i]);
}