Codeforces 629D Babaei and Birthday Cake(线段树优化dp)

来源:互联网 发布:幼儿园区域阅美工图片 编辑:程序博客网 时间:2024/06/04 08:53

题意:

n个蛋糕编号从小到大编号,j号蛋糕可以放在i号上面,当且仅当j的体积严格大于ii<j,问最终可得的最大蛋糕体积。

分析:

实质为求最长上升子序列问题,设dp[i]从头开始到第i位的最长子序列长度,可以想到O(n2)的做法,状态转移方程:

dp[i] = max(dp[j], j >= 0 && j < i && v[j] < v[i]) + v[i];

但是n可达1e5,这样做会超时。。。
那么如何快速的获取满足v[j]小于v[i]的最大的dp[j]呢?可以先将所有蛋糕体积进行离散化,用在排序后的数组的位置pos作为线段树下标,用线段树维护区间内最大的dp值,获取最大值时只需查询[0,pos)区间的dp最大值,时间复杂度为O(logn)
那么这样能否确保是按标号顺序进行叠放的呢?由于我们是从0n1进行遍历,排在后面的蛋糕,即使体积比当前蛋糕小,但是dp值并未更新,仍然为0,当前蛋糕是肯定不会选择放在它上面的~~~~

代码:

#include<cstdio>#include<cmath>#include<iostream>#include<algorithm>using namespace std;//[)const int maxm = 100005, maxn = maxm<<2,  INF = 0x3fffffff;#define pi acos(-1.0)typedef long long ll;ll dp[maxn];ll v[maxm], tv[maxm];void build(int k, int l, int r){    dp[k] = 0;    if(l == r - 1) return;    int mid = (l + r) / 2;    build(2 * k + 1, l, mid);    build(2 * k + 2, mid, r);}void update(int num, ll x, int k, int l, int r){    dp[k] = max(dp[k], x);    if(l == r - 1) return;    int mid = (l + r)/2;    if(mid > num)        update(num, x, k * 2 + 1, l, mid);    else        update(num, x, k * 2 + 2, mid, r);}ll query(int a, int b, int k, int l, int r){    if(a >= r||b <= l) return 0;    else if(a <= l && b >= r) return dp[k];    else{        int mid = (l + r) / 2;        ll ta = query(a, b, k * 2 + 1, l, mid);        ll tb = query(a, b, k * 2 + 2, mid, r);        return max(ta, tb);    }}int main (void){   int n;scanf("%d",&n);   int r, h;   for(int i = 0; i < n; i++){        scanf("%d%d",&r,&h);        tv[i] = v[i] =(ll) r * r * h;   }   sort(tv, tv + n);   int tot = unique(tv, tv + n) - tv;   build(0, 0, tot);   ll res = 0;   for(int i = 0; i < n; i++){        int pos = lower_bound(tv, tv + tot, v[i]) - tv;        ll tmp = query(0, pos, 0, 0, tot) + v[i];        res = max(res, tmp);        update(pos, tmp, 0, 0, tot);    }   printf("%.10lf\n", res * pi);   return 0;}
0 0