Codeforces Round #343 (Div. 2) D. Babaei and Birthday Cake（dp、BIT）

題意:

給定N≤105個蛋糕,編號爲1∼N,每個都有體積Vi
任意一個蛋糕都可以放在桌子上,對於蛋糕對(i,j),i可以放在j上,當且僅當i>j且Vi≥Vj
求能擺放的蛋糕的最大體積和

分析:

LIS模型的經典水題了
f[i]表示第i個蛋糕在最上面的最大體積和
f[i]=max{f[j]}+Vi,j<i且Vj<Vi
考慮用BIT優化一下這個轉移,離散化一下,記錄當前體積蛋糕在最上面的最大體積和,然後就可以O(logn)轉移了
ans=max{f[i]},時間復雜度爲O(nlogn)

代碼:

//
//  Created by TaoSama on 2016-02-23
//  Copyright (c) 2016 TaoSama. All rights reserved.
//
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <string>
#include <set>
#include <vector>

using namespace std;
#define pr(x) cout << #x << " = " << x << "  "
#define prln(x) cout << #x << " = " << x << endl
const int N = 1e5 + 10, INF = 0x3f3f3f3f, MOD = 1e9 + 7;

typedef long long LL;
int n;
LL v[N];

struct BIT {
    int n; LL b[N];
    void init(int _n) {
        n = _n;
        memset(b, 0, sizeof b);
    }
    void add(int i, LL v) {
        for(; i <= n; i += i & - i) b[i] = max(b[i], v);
    }
    LL sum(int i) {
        LL ret = 0;
        for(; i; i -= i & -i) ret = max(ret, b[i]);
        return ret;
    }
} bit;

int main() {
#ifdef LOCAL
    freopen("C:\\Users\\TaoSama\\Desktop\\in.txt", "r", stdin);
//  freopen("C:\\Users\\TaoSama\\Desktop\\out.txt","w",stdout);
#endif
    ios_base::sync_with_stdio(0);

    while(scanf("%d", &n) == 1) {
        vector<LL> xs;
        for(int i = 1; i <= n; ++i) {
            int r, h; scanf("%d%d", &r, &h);
            v[i] = 1LL * r * r * h;
            xs.push_back(v[i]);
        }
        sort(xs.begin(), xs.end());
        xs.resize(unique(xs.begin(), xs.end()) - xs.begin());

        LL ans = 0;
        bit.init(xs.size());
        for(int i = 1; i <= n; ++i) {
            int idx = lower_bound(xs.begin(), xs.end(), v[i]) - xs.begin() + 1;
            LL cur = bit.sum(idx - 1) + v[i];
            ans = max(ans, cur);
            bit.add(idx, cur);
        }
        printf("%.12f\n", ans * acos(-1));
    }
    return 0;
}