AtCoder Beginner Contest 221 E LEQ

最新推荐文章于 2024-09-15 12:49:55 发布

JXNU_SONG

最新推荐文章于 2024-09-15 12:49:55 发布

阅读量199

点赞数

分类专栏：数据结构思维文章标签： c++ c语言

本文链接：https://blog.csdn.net/qq_45863710/article/details/120604979

版权

思维同时被 2 个专栏收录

55 篇文章 1 订阅

订阅专栏

数据结构

35 篇文章 1 订阅

订阅专栏

这篇博客介绍了如何解决一个关于整数序列子序列计数的问题。通过变形正序对并使用动态规划技巧，实现了单点更新和查询操作，最终计算满足条件的子序列个数。文章涉及动态规划、二进制指数、前缀和等算法，并提供了C++代码实现。

摘要由CSDN通过智能技术生成

LEQ

https://atcoder.jp/contests/abc221/tasks/abc221_e

题目大意：给出一个长度为 $n$ 的整数序列 $a$ ，问有多少个子序列（无需连续） $a_1,a_2...a_k$ ，满足 $a_1≤a_k$ 。

那么我们发现可以将问题看成对于每个正序对去算一个值最后求和。那么我们这里正序对维护值的时候就不再是简单的加一了。假设 $a_i$ 作为正序对小的那一方，与 $a_i$ 可构成正序对的元素为 ${a_{j1},a_{j2}...a_{jk}\}$ ，那么其满足条件的子序列个数即为：
$2^{j_1-i-1}+2^{j_2-i-1}...+2^{j_k-i-1}=\frac{\sum\limits_{g=0}^{k}2^{j_g}}{2^{i+1}}$

所以我们发现我们只要对于求正序对去变个形，单点更新的是 $2^{下标}$ ，其它一致，最后再多除一个 $2^{i+1}$ 即算出对于 $a_i$ 作为 $a_1$ 的所有满足条件的答案，那么我们自然就可以遍历 $a_i$ 算出总答案。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 3e5 + 10;

const ll mod = 998244353;

int n, a[N];

ll er[N], inv[N], ans;

ll sum[N << 5];
int lson[N << 5], rson[N << 5], root, tot;

ll ksm(ll base, ll n) {
    ll ans = 1;
    while(n) {
        if (n & 1ll) ans = ans * base % mod;
        base = base * base % mod;
        n >>= 1ll;
    }
    return ans;
}

void init(int n) {
    er[0] = 1;
    for (int i = 1; i <= n; ++i) {
        er[i] = 2ll * er[i - 1] % mod;
    }
    inv[n] = ksm(er[n], mod - 2);
    for (int i = n - 1; i >= 0; --i) {
        inv[i] = inv[i + 1] * 2ll % mod;
    }
}

void update(int &rt, int l, int r, int pos, int idx) {
    if (!rt) rt = ++tot;
    if (l == r) {
        sum[rt] = (sum[rt] + er[idx]) % mod;
        return ;
    }
    int mid = (1ll * l + r) >> 1;
    if (pos <= mid) update(lson[rt], l, mid, pos, idx);
    else update(rson[rt], mid + 1, r, pos, idx);
    sum[rt] = (sum[lson[rt]] + sum[rson[rt]]) % mod;
}

ll query(int rt, int l, int r, int L, int R) {
    if (L <= l && r <= R) return sum[rt];
    int mid = (1ll * l + r) >> 1;
    ll ans = 0;
    if (mid >= L) ans = query(lson[rt], l, mid, L, R);
    if (mid < R) ans = (ans + query(rson[rt], mid + 1, r, L, R)) % mod;
    return ans;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    init(300001);
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
    }
    for (int i = n; i > 0; --i) {
        ll cur = query(root, 1, 1e9, a[i], 1e9) * inv[i + 1] % mod;
        ans = (ans + cur) % mod;
        update(root, 1, 1e9, a[i], i);
    }
    printf("%lld", ans);
}