首页 > 代码库 > ZOJ-3632 Watermelon Full of Water 线段树+DP

ZOJ-3632 Watermelon Full of Water 线段树+DP

暑假生活开始了,夏日炎炎,集训队想要每天都吃到西瓜。已知n天,每天商店提供一个西瓜,不同的西瓜可以供集训队吃不同的天数,也有不同的价格,问集训队想保证每天都能吃到西瓜的最小花费。

单个数100000,数组大小50000,因此需要用线段树优化。

对于每天的西瓜,不取则从最小值数组里取出当前最小值,取的话则是找出最小值+当天的西瓜价格,并且线段树更新后k天的最小费用。

dp[i][1]=min(dp[i-1][0],dp[i-1][1])+v[i];

dp[i][0]=query(i,i,1,n,1);

update(dp[i][1],i,min(n,i+d[i]-1),1,n,1);

由于最大值可能为50000*100000会超int,所以最大值inf要注意取大一点,比如1ll<<60

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <queue>
#include <ctime>
#include <vector>
#include <algorithm>

#define MAXN 100010
#define INF (1ll<<60)
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
ll v[MAXN];
ll d[MAXN];
ll dp[MAXN][3];
ll n;
ll sum[MAXN<<2],add[MAXN<<2];
void pushdown(ll rt){
    if(add[rt]!=INF){
   	 	add[rt<<1] = min(add[rt<<1],add[rt]);
        add[rt<<1|1] = min(add[rt<<1|1],add[rt]);
        sum[rt<<1] = min(sum[rt<<1],add[rt]);
        sum[rt<<1|1] = min(sum[rt<<1|1],add[rt]);
        add[rt] = INF;
    }
}
void pushup(ll rt){
    sum[rt] = min(sum[rt<<1],sum[rt<<1|1]);
}
void build(ll l,ll r,ll rt){
    sum[rt] = INF;
    add[rt] = INF;
    if(l==r)    return ;
    ll m = (l+r)>>1;
    build(lson);
    build(rson);
}
void update(ll c,ll L,ll R,ll l,ll r,ll rt){
    if(L<=l&&r<=R){
        add[rt] = min(add[rt],c);
        sum[rt] = min(c,sum[rt]);
        return ;
    }
    pushdown(rt);
    ll m = (l+r)>>1;
    if(L<=m)    update(c,L,R,lson);
    if(R>m)     update(c,L,R,rson);
    pushup(rt);
}
ll query(ll L,ll R,ll l,ll r,ll rt){
    if(L<=l&&r<=R){
        return sum[rt];
    }
    pushdown(rt);
    ll m = (l+r)>>1;
    ll ans = INF;
    if(L<=m)    ans = min(ans,query(L,R,lson));
    if(R>m)     ans = min(ans,query(L,R,rson));
    return ans;
}

int main()
{
	while(scanf("%lld",&n)!=EOF)
	{
		build(1,n,1);
		for(ll i=1;i<=n;i++)
		{
			scanf("%lld",&v[i]);
		}
		for(ll i=1;i<=n;i++)
		{
			scanf("%lld",&d[i]);
		}
		dp[1][0]=INF;
		dp[1][1]=v[1];
		update(v[1],1,min(n,d[1]),1,n,1);
		for(ll i=2;i<=n;i++)
		{
			dp[i][1]=min(dp[i-1][0],dp[i-1][1])+v[i];
			dp[i][0]=query(i,i,1,n,1);
			update(dp[i][1],i,min(n,i+d[i]-1),1,n,1);
		}
		printf("%lld\n",min(dp[n][1],dp[n][0]));
	}
	return 0;
} 


ZOJ-3632 Watermelon Full of Water 线段树+DP