洛谷P1156 垃圾陷阱[背包DP]

题目描述

卡门――农夫约翰极其珍视的一条Holsteins奶牛――已经落了到“垃圾井”中。“垃圾井”是农夫们扔垃圾的地方，它的深度为D(2<=D<=100)英尺。

卡门想把垃圾堆起来，等到堆得与井同样高时，她就能逃出井外了。另外，卡门可以通过吃一些垃圾来维持自己的生命。

每个垃圾都可以用来吃或堆放，并且堆放垃圾不用花费卡门的时间。

假设卡门预先知道了每个垃圾扔下的时间t(0< t<=1000)，以及每个垃圾堆放的高度h(1<=h<=25)和吃进该垃圾能维持生命的时间f(1<=f<=30)，要求出卡门最早能逃出井外的时间，假设卡门当前体内有足够持续10小时的能量，如果卡门10小时内没有进食，卡门就将饿死。

输入输出格式

输入格式：

第一行为2个整数，D 和 G (1 <= G <= 100)，G为被投入井的垃圾的数量。

第二到第G+1行每行包括3个整数：T (0 < T <= 1000)，表示垃圾被投进井中的时间；F (1 <= F <= 30)，表示该垃圾能维持卡门生命的时间；和 H (1 <= H <= 25)，该垃圾能垫高的高度。

输出格式：

如果卡门可以爬出陷阱，输出一个整表示最早什么时候可以爬出；否则输出卡门最长可以存活多长时间。

输入输出样例

输入样例#1：

输出样例#1：

说明

[样例说明]

卡门堆放她收到的第一个垃圾：height=9；

卡门吃掉她收到的第二个垃圾，使她的生命从10小时延伸到13小时；

卡门堆放第3个垃圾，height=19；

卡门堆放第4个垃圾，height=20。

一开始先想f[i][j]表示前i个高度j的最大剩余时间

遇到一个垃圾i可以使到j有剩余就输出a[i].t

否则从f[i][0]+a[i].t里选最优

注意的是0那一刻卡门还没有死，所以初始化-1

//

//  main.cpp

//  luogu1156

//

//  Created by Candy on 10/11/2016.

//  Copyright © 2016 Candy. All rights reserved.

//

#include <iostream>

#include <cstdio>

#include <algorithm>

#include <cstring>

using namespace std;

const int M=3005,N=105;

inline int read(){

    char c=getchar();int x=0,f=1;

    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}

    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}

    return x*f;

}

int d,H,n,mp[N];

struct data{

    int t,w,h;

    bool operator <(const data &r)const{return t<r.t;}

}a[N];

int f[N][M];

bool dp(){

    memset(f,-1,sizeof(f));

    f[0][0]=10;

    a[0].t=0;

    for(int i=1;i<=n;i++)

        for(int j=0;j<=H;j++){

            int t=a[i].t-a[i-1].t;

            if(f[i-1][j]-t>=0)

                f[i][j]=f[i-1][j]-t+a[i].w;

            if(j-a[i].h>=0&&f[i-1][j-a[i].h]-t>=0)

                f[i][j]=max(f[i][j],f[i-1][j-a[i].h]-t);

            //printf("f %d %d %d\n",i,j,f[i][j]);

            if(j>=d&&f[i][j]>=0){printf("%d\n",a[i].t);return true;}

        }

    return false;

}

int main(int argc, const char * argv[]) {

    d=read();n=read();

    for(int i=1;i<=n;i++){

        a[i].t=read();a[i].w=read();a[i].h=read();

        H+=a[i].h;

    }

    sort(a+1,a+1+n);

    if(!dp()){

        int mx=10;

        //for(int i=n;i>=1;i--) if(f[i][0]>=0) mx=max(mx,f[i][0]+a[i].t);

        for(int i=1;i<=n;i++){

            if(mx<a[i].t) break;

            mx+=a[i].w;

        }

        printf("%d",mx);

    }

    return 0;

}

还有一种写法，f[i][j]表示最长活的时间

顺便滚动数组+刷表法要快一些

//

//  main.cpp

//  luogu1156

//

//  Created by Candy on 10/11/2016.

//  Copyright © 2016 Candy. All rights reserved.

//

#include <iostream>

#include <cstdio>

#include <algorithm>

#include <cstring>

using namespace std;

const int N=105,M=3005;

inline int read(){

    char c=getchar();int x=0,f=1;

    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}

    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}

    return x*f;

}

int d,H,n,mp[N];

struct data{

    int t,w,h;

    bool operator <(const data &r)const{return t<r.t;}

}a[N];

int f[M];

bool dp(){

    f[0]=10;

    for(int i=1;i<=n;i++)

        for(int j=d;j>=0;j--) if(f[j]>=a[i].t){

            f[j+a[i].h]=max(f[j],f[j+a[i].h]);

            f[j]+=a[i].w;

            if(j+a[i].h>=d){printf("%d",a[i].t);return true;}

        }

    return false;

}

int main(int argc, const char * argv[]) {

    d=read();n=read();

    for(int i=1;i<=n;i++){

        a[i].t=read();a[i].w=read();a[i].h=read();

        H+=a[i].h;

    }

    sort(a+1,a+1+n);

    if(!dp()) printf("%d",f[0]);

    return 0;

}

貌似f[i]表示时间i的最大高度j也可以