首页 > 技术文章 > P1156 垃圾陷阱(01背包)

oneman233 2019-09-04 22:02 原文

明显的背包,但是不看题解真想不出来怎么保存状态。

首先一点,他一定会在拿到垃圾的第一时间处理垃圾,这样才能更快地逃出井,除非他逃不出去。

我首先想dp[i][j]表示高度为i血量为j时候的最少天数,但是发现没法控制物品的选择时间,实际上应该用第一维表示物品编号,首先把物品按照下落时间排序。

先考虑dp[i][j]是第i个物品血量为j时候的最大高度,暂时不考虑时间的限制,搞一下方程:

dp[i][j]=max(dp[i-1][j]+h[i],dp[i-1][j-f[i]]);

两种决策,要么垫高要么吃了。

大概有点意思了,但是发现血量的大小关系挺混乱的,不好枚举范围。

换个思路,dp[i][j]表示前i个物品处理之后在j高度时候的最大血量,高度是有上限的,枚举范围就定下来了。

方程也比较好想:dp[i][j]=max(dp[i-1][j-h[i]],dp[i-1][j]+f[i]);

注意上式中第一维代表拿来垫高,第二维代表拿来吃了。

下面要考虑如何处理时间的限制:

对任意一个状态,不死的条件是:dp[i-1][j]>=t[i]

这意味着在i-1个物品的状态下剩余血量为j能存活到这个物品落下的时间。

具体的转移方程如下:

re(i,1,g){
        re(j,0,d){
            if(dp[i-1][j]>=o[i].t)
                dp[i][j]=max(dp[i][j],dp[i-1][j]+o[i].f);
            if(j>=o[i].h&&dp[i-1][j-o[i].h]>=o[i].t)
                dp[i][j]=max(dp[i][j],dp[i-1][j-o[i].h]);
        }
    }

dp的初始值赋为最小,然后设定dp[0][0]=10,最后dp结束以后全部遍历一遍dp数组,更新答案。

如果最大高度到达了d,说明在垫高该物品后可以逃出井,输出答案为该物品下落的时间,否则输出最大血量即可。

完整代码如下:

#include <bits/stdc++.h>
#define int long long
#define sc(a) scanf("%lld",&a)
#define scc(a,b) scanf("%lld %lld",&a,&b)
#define sccc(a,b,c) scanf("%lld %lld %lld",&a,&b,&c)
#define schar(a) scanf("%c",&a)
#define pr(a) printf("%lld",a)
#define fo(i,a,b) for(int i=a;i<b;++i)
#define re(i,a,b) for(int i=a;i<=b;++i)
#define rfo(i,a,b) for(int i=a;i>b;--i)
#define rre(i,a,b) for(int i=a;i>=b;--i)
#define prn() printf("\n")
#define prs() printf(" ")
#define mkp make_pair
#define pii pair<int,int>
#define pub(a) push_back(a)
#define pob() pop_back()
#define puf(a) push_front(a)
#define pof() pop_front()
#define fst first
#define snd second
#define frt front()
#define bak back()
#define mem0(a) memset(a,0,sizeof(a))
#define memmx(a) memset(a,0x3f3f,sizeof(a))
#define memmn(a) memset(a,-0x3f3f,sizeof(a))
#define debug
#define db double
#define yyes cout<<"YES"<<endl;
#define nno cout<<"NO"<<endl;
using namespace std;
typedef vector<int> vei;
typedef vector<pii> vep;
typedef map<int,int> mpii;
typedef map<char,int> mpci;
typedef map<string,int> mpsi;
typedef deque<int> deqi;
typedef deque<char> deqc;
typedef priority_queue<int> mxpq;
typedef priority_queue<int,vector<int>,greater<int> > mnpq;
typedef priority_queue<pii> mxpqii;
typedef priority_queue<pii,vector<pii>,greater<pii> > mnpqii;
const int maxn=500005;
const int inf=0x3f3f3f3f3f3f3f3f;
const int MOD=100000007;
const db eps=1e-10;
int qpow(int a,int b){int tmp=a%MOD,ans=1;while(b){if(b&1){ans*=tmp,ans%=MOD;}tmp*=tmp,tmp%=MOD,b>>=1;}return ans;}
int lowbit(int x){return x&-x;}
int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
int mmax(int a,int b,int c){return max(a,max(b,c));}
int mmin(int a,int b,int c){return min(a,min(b,c));}
void mod(int &a){a+=MOD;a%=MOD;}
bool chk(int now){}
int half(int l,int r){while(l<=r){int m=(l+r)/2;if(chk(m))r=m-1;else l=m+1;}return l;}
int ll(int p){return p<<1;}
int rr(int p){return p<<1|1;}
int mm(int l,int r){return (l+r)/2;}
int lg(int x){if(x==0) return 1;return (int)log2(x)+1;}
bool smleql(db a,db b){if(a<b||fabs(a-b)<=eps)return true;return false;}
db len(db a,db b,db c,db d){return sqrt((a-c)*(a-c)+(b-d)*(b-d));}
bool isp(int x){if(x==1)return false;if(x==2)return true;for(int i=2;i*i<=x;++i)if(x%i==0)return false;return true;}

int d,g,dp[1005][1005];
struct node{
    int t,f,h;
}o[maxn];

bool cmp(node a,node b){
    return a.t<b.t;
}

signed main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    cin>>d>>g;
    re(i,1,g) cin>>o[i].t>>o[i].f>>o[i].h;
    sort(o+1,o+1+g,cmp);
    memmn(dp);
    dp[0][0]=10;
    re(i,1,g){
        re(j,0,d){
            if(dp[i-1][j]>=o[i].t)
                dp[i][j]=max(dp[i][j],dp[i-1][j]+o[i].f);
            if(j>=o[i].h&&dp[i-1][j-o[i].h]>=o[i].t)
                dp[i][j]=max(dp[i][j],dp[i-1][j-o[i].h]);
        }
    }
    int mh=-inf,mt=-inf,p;
    re(i,1,g){
        re(j,0,d){
            if(dp[i][j]>=o[i].t)
                mh=max(mh,j);
            mt=max(mt,dp[i][j]);
        }
        if(mh==d)
            {p=i;break;}
    }
    if(mh==d) cout<<o[p].t;
    else cout<<mt;
    return 0;
}

 

推荐阅读