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; }