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

题意:

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
", res * pi);
   return 0;
}
原文地址:https://www.cnblogs.com/Tuesdayzz/p/5758765.html