Codeforces Round #436 E. Fire(背包dp+输出路径)

题意:失火了,有n个物品,每个物品有价值pi,必须在时间di前(小于di)被救,否则就要被烧毁。救某个物

    品需要时间ti,问最多救回多少价值的物品,并输出救物品的顺序。

 

 

Examples

 

Input
3
3 7 4
2 6 5
3 7 6

 

Output
11
2
2 3

 

Input
2
5 6 1
3 3 5

 

Output

 

1
1
1

思路:有点像一个背包,dp数组记录的是当前时间所能获得的最大价值,转移方程dp[j]=max(dp[j],dp[j-t[i].ti]+t[i].w)
    path[i][j]表示到i号物品j时间的状态时救的物品编号(这里是从0开始)
    emmmmm 具体的在代码中加以注释。

代码:

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn=2010;
struct node{
    int ti,en,w,op;
}t[110];
int dp[maxn],path[110][maxn],a[110];

bool cmp(node x,node y){
    if(x.en!=y.en)return x.en<y.en;
    else return x.ti<y.ti;
}


int main(){
    int n;
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>t[i].ti>>t[i].en>>t[i].w;
        t[i].op=i+1;
        //op为每个物品的标号,题目要求从1开始
    }
    memset(dp,0,sizeof(dp));
    memset(path,-1,sizeof(path));//开始全部置为-1
    sort(t,t+n,cmp);
    int ma=0;
    for(int i=0;i<n;i++){
        ma=max(ma,t[i].en);
        if(i!=0){
            for(int j=0;j<t[i].ti;j++)path[i][j]=path[i-1][j];
            //时间小于所需的营救时间,那么就不救,等于上一状态
        }
        for(int j=t[i].en-1;j>=t[i].ti;j--){
            if(dp[j]<dp[j-t[i].ti]+t[i].w){
                dp[j]=dp[j-t[i].ti]+t[i].w;
                path[i][j]=i;
                //如果救,那么就更新dp数组以及path数组为该物品
            }
            else if(i>0){
                path[i][j]=path[i-1][j];
                //如果不救,那么还是等于上一状态
            }
        }
    }
    int sum=0,temp=0;
    for(int i=0;i<=ma;i++){
        if(dp[i]>sum){
            sum=dp[i];
            temp=i;
            //取出最大价值以及在最大价值的情况下,救完最后一个物品的时间
        }
    }
    int k=n-1,c=0;
    while(temp>0){
        a[c++]=t[path[k][temp]].op;
        temp-=t[path[k][temp]].ti;
        k--;
        k=path[k][temp];
        //反着记录下路径
    }
    cout<<sum<<endl<<c<<endl;
    for(int i=c-1;i>=0;i--){
        if(i!=c-1)cout<<' ';
        cout<<a[i];
        //倒叙输出
    }
    cout<<endl;
    return 0;
}

 

posted @ 2017-10-07 16:39  ljy3268  阅读(224)  评论(0编辑  收藏  举报