花朵选拔(Flowers) 题解

思路分析

按题目要求使用对应数据结构与算法模拟/递推求解，注意数据范围边界与时间复杂度。

参考代码

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n;
    if (!(cin >> n)) return 0;
    vector<int> h(n);
    vector<long long> a(n);
    for (int i = 0; i < n; ++i) cin >> h[i];
    for (int i = 0; i < n; ++i) cin >> a[i];
    vector<long long> bit(n + 1, 0);
    auto upd = [&](int idx, long long v) {
        for (int i = idx; i <= n; i += i & -i) bit[i] = max(bit[i], v);
    };
    auto qry = [&](int idx) {
        long long res = 0;
        for (int i = idx; i > 0; i -= i & -i) res = max(res, bit[i]);
        return res;
    };
    long long ans = 0;
    for (int i = 0; i < n; ++i) {
        long long best = qry(h[i] - 1) + a[i];
        upd(h[i], best);
        ans = max(ans, best);
    }
    cout << ans << "\n";
    return 0;
}