既然是用dp求第K优解 原来一维的数组必然变成了二维 用来存储是第几优的解
这里用了一个n*n*n的算法做的……貌似也没办法优化了 分析一下做法 既然说 dp[j][m]也是根据逐次比较得来的 那么需要j固定时 逐个记录可能粗线的dp值
/************ hdu2639 2015.10.20 109MS 5548K 1226B ************/ #include <iostream> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; int dp[1009][1009]; int value[109],cost[109]; int n,k,v,t; int a[1009],b[1009]; int main() { while(~scanf("%d",&t)) { while(t--) { scanf("%d%d%d",&n,&v,&k); for(int i=1;i<=n;i++) scanf("%d",&value[i]); for(int i=1;i<=n;i++) scanf("%d",&cost[i]); memset(dp,0,sizeof(dp)); for(int i=1;i<=n;i++) for(int j=v;j>=cost[i];j--) { for(int m=1;m<=k;m++) { a[m]=dp[j][m]; b[m]=dp[j-cost[i]][m]+value[i]; } int x=1,y=1,w=1; a[k+1]=-1;b[k+1]=-1; while((w<=k)&&(x<=k||y<=k)) { if(a[x]<b[y]) dp[j][w]=b[y++]; else dp[j][w]=a[x++]; if(w==1||dp[j][w-1]!=dp[j][w]) w++; } } printf("%d\n",dp[v][k]); } } return 0; }