zoj 3632 Watermelon Full of Water

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=4778

 

题目大意：

暑假有n（1~50000）天,第i天西瓜的价格是p[i]，可以吃d[i]天，如果在第i天买了个西瓜，那么旧的西瓜要扔掉，开始吃新西瓜。

问每天都能吃到西瓜的最小花费。

 

 

题目思路：

p[i]表示第i天买喜欢的花费，d[i]表示第i天买的西瓜能吃的天数。

dp[i],表示以i为起点，最少要花费多少钱才能使i~n每天都有西瓜吃。

状态转移很显然dp[i]= (i+d[i]>n)? p[i] : min{dp[i+1]~dp[i+d[i]]}+p[i]。

很明显在取dp[i+1]~dp[i+d[i]]的过程显然不可以从i+1 ~ i+d[i]扫一遍，这样复杂度O（n*n），会超时。

所以最直接的方法就是用线段树优化。

 

代码：

#include <stdlib.h>

#include <string.h>

#include <stdio.h>

#include <ctype.h>

#include <math.h>

#include <stack>

#include <queue>

#include <map>

#include <set>

#include <vector>

#include <string>

#include <iostream>

#include <algorithm>

using namespace std;



#define ll long long

#define ls rt<<1

#define rs ls|1

#define lson l,mid,ls

#define rson mid+1,r,rs

#define middle (l+r)>>1

#define clr_all(x,c) memset(x,c,sizeof(x))

#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))

#define eps (1e-8)

#define MOD 1000000007

#define INF 0x3f3f3f3f

#define PI (acos(-1.0))

#pragma comment(linker, "/STACK:102400000,102400000")

template <class T> T _max(T x,T y){return x>y? x:y;}

template <class T> T _min(T x,T y){return x<y? x:y;}

template <class T> T _abs(T x){return (x < 0)? -x:x;}

template <class T> T _mod(T x,T y){return (x > 0)? x%y:((x%y)+y)%y;}

template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;}

template <class T> void getmax(T& x,T y){x=(y > x)? y:x;}

template <class T> void getmin(T& x,T y){x=(x<0 || y<x)? y:x;}

int TS,cas=1;

const int M=50000+5;

ll n,p[M],d[M],mmin[M<<2],MAX;

void build(int l,int r,int rt){

	mmin[rt]=MAX;

	if(l==r) return;

	int mid=middle;

	build(lson),build(rson);

}

void pushUp(int rt){

	mmin[rt]=_min(mmin[ls],mmin[rs]);

}

void update(int l,int r,int rt,int p,ll c){

	if(l==r){mmin[rt]=c;return;}

	int mid=middle;

	if(p<=mid) update(lson,p,c);

	else update(rson,p,c);

	pushUp(rt);

}

ll query(int l,int r,int rt,int L,int R){

	if(L<=l && r<=R) return mmin[rt];

	int mid=middle;

	if(R<=mid) return query(lson,L,R);

	else if(mid<L) return query(rson,L,R);

	else return _min(query(lson,L,mid),query(rson,mid+1,R));

}



void run(){

	int i,j;

	MAX=1;

	for(i=1;i<=n;i++) scanf("%lld",&p[i]),MAX+=p[i];

	for(i=1;i<=n;i++) scanf("%lld",&d[i]);

	ll ret,tmp;

	build(1,n,1);

	for(i=n;i>=1;i--){

		int l=i+1,r=i+d[i];

		if(r>n) ret=p[i];

		else ret=query(1,n,1,i+1,i+d[i])+p[i];

		update(1,n,1,i,ret);

	}

	printf("%lld\n",ret);

}



void preSof(){

}



int main(){

    //freopen("input.txt","r",stdin);

    //freopen("output.txt","w",stdout);

    preSof();

    //run();

    while((~scanf("%d",&n))) run();

    //for(scanf("%d",&TS);cas<=TS;cas++) run();

    return 0;

}