CF 629-D Babaei and Birthday Cake //dp+BIT

题目链接

http://codeforces.com/problemset/problem/629/D

题意

很多蛋糕，每个蛋糕有高和半径，i号蛋糕能放在j号蛋糕的上面，当且仅当$i>j$且$v[i]>v[j]$($v[i]$指i号蛋糕的体积)，让求最后可以得到的蛋糕的最大体积。

解析

类比着LIS，很容易想到用动态规划求解，定义$dp[i]$为以i号蛋糕结尾通过组合的最大体积。

则:$dp[i]=max(dp[j])+v[i](i>j且v[i]>v[j])$

由于n较大，不能直接按照求LIS的普通做法求解，需要优化，所以可以用线段树/树状数组进行优化，求解$dp[i]$需要去前面查询能满足的最大的$dp[j]$，此处进行优化，以树状数组为例，维护这个$dp[j]$，每次查询去前面找最大值即可。

还有一点就是，由于圆柱体积为:$v=h\ast \Pi \ast r^2$，可以只求出$h\ast r^2$，然后进行离散化，方便树状数组维护，但原来求的$h\ast r^2$还要保留，因为在过程中还需要用到这些值。时间复杂度:$O(nlo{g}_n)$

参考代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
const int inf=0x3f3f3f3f;
const int maxn=1e5+7;
const double pi=acos(-1.0);
int n,m;
ll a[maxn],b[maxn],c[maxn];
ll dp[maxn];
void discretization(){
    sort(b+1,b+1+n);
    int len=unique(b+1,b+1+n)-b-1;
    for(int i=1;i<=n;i++){
        a[i]=lower_bound(b+1,b+1+len,a[i])-b;
    }
}
struct BIT{
    ll bit[maxn];
    inline void init(){ for(int i=1;i<=n;i++)bit[i]=0; }
    inline int lowbit(int x){ return x&(-x); }
    inline void update(int x,ll val){ for(int i=x;i<=n;i+=lowbit(i)) bit[i]=max(bit[i],val); }
    inline ll query(int x,ll res=0){ for(int i=x;i;i-=lowbit(i))res=max(bit[i],res); return res; }
}s;
int main()
{
    while(~scanf("%d",&n)){
        for(int i=1;i<=n;i++){
            ll r,h;scanf("%lld%lld",&r,&h);
            a[i]=r*r*h;//离散化数组
            b[i]=a[i];//进行离散化的辅助数组
            c[i]=a[i];//保留原体积
        }
        discretization();
        s.init();
        ll ans=0;
        for(int i=1;i<=n;i++){
            dp[i]=s.query(a[i]-1)+c[i];
            s.update(a[i],dp[i]);//树状数组维护dp[j]
            ans=max(ans,dp[i]);
        }
        double sum=(double)ans*pi;
        printf("%.9f\n",sum);
    }
    return 0;
}