问题
https://vjudge.net/problem/UVA-607
分析
这是一道线性dp,多出了一个需求,就是对应最少课程数的最少DI值,所以可以先求出最少课程数,顺便求对应的最少DI值,每次更新最少课程数时,或者相等时,考虑di值的变化
代码
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <map>
#include <string>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long LL;
const int maxn=1000+5,Inf=0x3f3f3f3f;
//使用线性dp就可以了,不需要使用二维的,二维的会很费时间
//dp[i]表示讲完了i个topic需要的最少课程数
//di[i]表示对应dp[i]的课程数的最小DI
int n,L,C,lec[maxn],tot[maxn],dp[maxn],di[maxn],kase=0,t;
inline int cost(int x){
if(x==0) return 0;
return (x<=10)?-C:(x-10)*(x-10);
}
int main(void){
while(scanf("%d",&n)==1 && n){
scanf("%d%d",&L,&C);
for(int i=1;i<=n;++i){
scanf("%d",&lec[i]);
tot[i]=tot[i-1]+lec[i];
}
dp[0]=di[0]=0;
for(int i=1;i<=n;++i){
dp[i]=di[i]=Inf;
for(int j=1;j<=i;++j){
if(tot[i]-tot[i-j]>L) break;
if(dp[i-j]+1<dp[i] || (dp[i-j]+1==dp[i] && cost(L-tot[i]+tot[i-j])+di[i-j]<di[i])){
dp[i]=dp[i-j]+1;
di[i]=cost(L-tot[i]+tot[i-j])+di[i-j];
}
}
}
if(kase) printf("\n");
printf("Case %d:\nMinimum number of lectures: %d\nTotal dissatisfaction index: %d\n",++kase,dp[n],di[n]);
}
return 0;
}