UVA - 607 Scheduling Lecture 线性dp

问题

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;
}