博客
关于我
P1637 三元上升子序列 (权值线段树优化dp)
阅读量:808 次
发布时间:2019-03-25

本文共 1940 字,大约阅读时间需要 6 分钟。

这道题的解法采用了线段树技术,通过两次预处理将问题的时间复杂度优化到了O(n log n)。首先,我们预处理每个元素的前置信息d1[i],表示前i个元素中小于a[i]的元素个数。接着,我们预处理满足三元上升条件的序列对数。

问题分析

题目要求在数组中找出所有满足三元递增条件的情况。我们需要一个高效算法来处理大规模数据。通过动态规划直接求解会导致时间复杂度过高,因此采用线段树来优化。

方法解法

为了优化计算过程,我们使用线段树进行预处理:

  • 第一轮预处理:插入数组元素到线段树中,计算每个元素前面的d1[i]值(小于当前元素的元素个数)。
  • 第二轮预处理:对每个元素,查询满足条件的后续元素个数,计算d2[i](满足三元递增的对数)。
  • 线段树维护两层信息:区间内元素的数量和d1的累积和。通过两轮预处理,实现高效查询和更新。

    解决代码

    using namespace std;
    #define ll long long
    const int maxn = 1e5 + 7;
    const ll inf = 34359738370;
    ll tree[maxn * 100];
    int root = 1, cnt = 1;
    int d1[maxn], d2[maxn];
    int n, a[maxn];
    ll ans = 0;
    void pushup(int rt) {
    tree[rt] = tree[lc[rt]] + tree[rc[rt]];
    return;
    }
    void updata(int &rt, ll l, ll r, ll x, int v) {
    if (!rt) rt = ++cnt;
    if (l == r) {
    tree[rt] += v;
    if (tree[rt] < 0) tree[rt] = 0;
    return;
    }
    ll mid = (l + r) >> 1;
    if (x <= mid) updata(lc[rt], l, mid, x, v);
    else updata(rc[rt], mid + 1, r, x, v);
    pushup(rt);
    }
    int query(int rt, ll l, ll r, ll vl, ll vr) {
    if (!rt || r < vl || l > vr) return 0;
    if (vl <= l && r <= vr) return tree[rt];
    ll mid = (l + r) >> 1;
    return query(lc[rt], l, mid, vl, vr) + query(rc[rt], mid + 1, r, vl, vr);
    }
    int main() {
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) {
    scanf("%d", a + i);
    }
    updata(root, 0, inf, a[1], 1);
    for (int i = 2; i <= n; i++) {
    d1[i] = query(root, 0, inf, 0, a[i] - 1);
    updata(root, 0, inf, a[i], 1);
    }
    for (int i = 1; i <= n; i++)
    updata(root, 0, inf, a[i], -maxn);
    for (int i = 2; i <= n; i++) {
    ans += query(root, 0, inf, 0, a[i] - 1);
    updata(root, 0, inf, a[i], d1[i]);
    }
    printf("%lld\n", ans);
    return 0;
    }

    代码解释

  • 线段树结构:定义了线段树的根节点和相关节点,用于存储区间信息。
  • pushup函数:用于上推节点信息,确保子节点数据正确合并。
  • updata函数:更新线段树中的节点信息,插入或删除元素时的处理。
  • 查询函数:查找满足条件的区间,快速得到d1和d2的值。
  • 预处理:插入元素,计算d1值后再预处理d2值,计算总对数。
  • 结果输出:最后打印满足三元递增的对数总和。
  • 通过上述步骤,我们可以高效解决三元递增子序列的问题。

    转载地址:http://wkjyk.baihongyu.com/

    你可能感兴趣的文章
    Vue3+element plus+sortablejs实现table列表拖拽
    查看>>
    Nokia5233手机和我装的几个symbian V5手机软件
    查看>>
    non linear processor
    查看>>
    Non-final field ‘code‘ in enum StateEnum‘
    查看>>
    none 和 host 网络的适用场景 - 每天5分钟玩转 Docker 容器技术(31)
    查看>>
    None还可以是函数定义可选参数的一个默认值,设置成默认值时实参在调用该函数时可以不输入与None绑定的元素...
    查看>>
    NoNodeAvailableException None of the configured nodes are available异常
    查看>>