bzoj1133: [POI2009]Kon

bzoj1133: [POI2009]Kon

链接

https://www.lydsy.com/JudgeOnline/problem.php?id=1133

思路

f[i][k]表示前i个，选了k个，其中必选i的最大值
f[i][k]=f[j][k-1]+贡献
这个贡献就是j到i之间的边界碰到i的人数(可以预处理个前缀和)
复杂度O(nnk)

错误

貌似输出方案错了还是神马的,90分
不过bzoj数据都过了呀，bzoj还会改数据吗

错误代码

#include <bits/stdc++.h>
using namespace std;
const int N=607,M=57;
int read() {
	int x=0,f=1;char s=getchar();
	for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
	for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
	return x*f;
}
int n,K,f[N][M],sum[N][N],frm[N][M];
int stak[N],top;
int main() {
	n=read(),K=read();
	for(int i=1;i<=n;++i)
		for(int j=i+1;j<=n;++j)
			sum[i][j]=sum[i][j-1]+read();
	for(int k=1;k<=K;++k) for(int i=1;i<=n;++i){
		int tot=0;
		for(int j=i;j>=1;--j) {
			tot+=sum[j][n]-sum[j][i];
			if(f[i][k]==f[j-1][k-1]+tot) frm[i][k]=min(frm[i][k],j-1); 
			if(f[i][k]<f[j-1][k-1]+tot) {
				f[i][k]=f[j-1][k-1]+tot;
				frm[i][k]=j-1;
			}
		}
	}
	int ma=0,x_x=0;
	for(int i=1;i<=n;++i) {
		if(ma<f[i][K]) ma=f[i][K],x_x=i;
	}
	while(K) {
		stak[++top]=x_x;
		x_x=frm[x_x][K--];
	}
	while(top--) printf("%d ",stak[top+1]);
	return 0;
}