题目大意:给定n个站点,每个人都会在某个站点上车并在之后的某个站点下车,查票员可以在两个站点之间查票,问查票k次最多查到多少人
壮哉我大轻音部(误
令f[i][j]表示当前在第i个点和第i+1个点之间查票,已经查了j次的最大收益
枚举上一次查票的位置,统计比上一次能多查出来的人数即可
时间复杂度O(kn^2) 输出方案记录一下上一次查票的位置即可
由于没有SPJ所以要输出字典序最小的方案
注意有组全零的数据
3 2
0 0 0
0 0
0
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define M 610 using namespace std; int n,k,ans,ans_pos; int a[M][M],f[M][55],from[M][55]; //f[i][j]表示当前在第i个点和第i+1个点之间查票,已经查了j次的最大收益 int output[55]; int Get_Sum(int x1,int y1,int x2,int y2) { return a[x2][y2]-a[x1-1][y2]-a[x2][y1-1]+a[x1-1][y1-1]; } int main() { int i,j,k; cin>>n>>::k; for(i=1;i<n;i++) for(j=i+1;j<=n;j++) scanf("%d",&a[i][j]); for(i=1;i<=n;i++) for(j=1;j<=n;j++) a[i][j]+=a[i-1][j]; for(i=1;i<=n;i++) for(j=1;j<=n;j++) a[i][j]+=a[i][j-1]; memset(f,0xef,sizeof f); f[0][0]=0; for(i=1;i<n;i++) for(j=1;j<=::k;j++) { for(k=0;k<i;k++) { int temp=Get_Sum(k+1,i+1,i,n); if(f[k][j-1]+temp>f[i][j]) { f[i][j]=f[k][j-1]+temp; from[i][j]=k; } } if(j==::k&&f[i][j]>ans) ans=f[i][j],ans_pos=i; } int temp=ans_pos; for(i=::k;i;i--) { output[i]=temp; temp=from[temp][i]; } for(i=1;i<=::k;i++) printf("%d%c",output[i]," \n"[i==::k]); return 0; }
时间: 2024-10-08 07:31:05