Luogu 5010 HMR的LIS Ⅲ 题解

最新推荐文章于 2024-08-15 22:41:27 发布

winsunboy

最新推荐文章于 2024-08-15 22:41:27 发布

阅读量660

点赞数 7

分类专栏：题解文章标签：算法动态规划

本文链接：https://blog.csdn.net/winsunboy/article/details/140935372

版权

题解专栏收录该内容

5 篇文章 0 订阅

订阅专栏

Luogu 5010 HMR的LIS Ⅲ

题意简化

给定长度为 $n$ 的序列 $a_1, a_2, \dots, a_n$ ，以及常数 $L$ 、 $R$ 。要求构造序列 $b$ ，满足（设 $b$ 序列的长度为 $m$ ）：

$\forall 1 \le i \le m$ ， $\le b_i \le n$ ；
$\forall 1 \le i < m$ ， $b_i < b_{i+1}$ ;
$\forall 1 \le i < m$ ， $L < a_{b_{i+1}} - a_{b_i} < R$ ；

并在此基础上使得长度 $m$ 最大。

然而，在保证长度最大的前提下，仍然有许多中可能的构造方案。给定 $k$ ，输出字典序第 $k$ 小的构造方案。

对于 $100\%$ 的数据：

$\le n \le 5 \times 10^5$ ， $\le k \le 10^{13}$ ；
$\le |L|, |R| \le 10^9 + 1$ ， $R - L > 1$ ， $\le a_i \le 10^9$ ；
保证数据合法且有解。

思路

同样是线段树优化 dp，但是我感觉这题比 NOIP2023 T4 天天爱打卡简单。往期回顾

$a_i$ 这么大，存线段树之前需要离散化，~~这是个常识吧~~。

$k$ 这么大，说明方案数可能很大，记得用 long long 来存，~~这也是个常识吧~~。

正序倒序都可以把 $m$ 求出来，但是倒序在构造方案的时候更方便。

设 $f_i$ 表示 $b_1 = i$ 时 $m$ 的最大值， $c_i$ 表示此时的方案数。

要求出字典序第 $k$ 小的方案。

这个东西怎么讲呢？如果能够从位置 $i$ 转移到前一个已选定的状态，如果 $\le c_i$ ，也就是说前 $c_i$ 个方案全部都要选 $i$ ，那么第 $k$ 个也一定要选 $i$ 。

如果 $k > c_i$ ，那么排除了这 $c_i$ 个方案之后，去到下一个可行的位置找第 $k - c_i$ 个方案。

而我们这一个从前往后确定方案的过程，恰好是按照字典序从小到大进行的，所以可行且方便。

for (int i = 1, lst = 0; i <= n && ans; ++i) {
    if (f[i] == ans && (!lst || (L+a[lst]<a[i] && R+a[lst]>a[i]))) { //能够转移
        if (k > c[i]) k -= c[i];
        else {
            printf("%d ", i);
            lst = i;
            --ans;
        }
    }
}

具体来讲讲 怎样 dp。

设 $h (x)$ 表示离散化后序号为 $x$ 的值，原来是多少。

考虑一个问题：如果当前为 $a_i$ ，那么能够转移过来的 $a_j$ 应该在哪个区间？

对 $L < a_j - a_i < R$ 进行变形，有 $L + a_i < a_j < R + a_i$ 。

开区间。。。没关系。。。

二分查找 $h(l) > L + a_i$ ， $l$ 的最小值； $h(r) < R + a_i$ ， $r$ 的最大值。如果区间 $[l, r]$ 合法，到线段树中查找这个区间的 $f_{max}$ 以及总共的方案数 $c^{'}$ ，则 $f_i = f_{max} + 1$ 。

如果 $f_{max} \ne 0$ ，那么 $c_i = c'$ ；否则 $c_i = 1$ 。

（当然如果连这个区间都不合法的话，显然 $f_i = 0$ ， $c_i = 1$ 。）

然后就把当前的情况拿去更新线段树 $h(a_i)$ 位置就行了。

for (int i = n; i >= 1; --i) {
    int l = upper_bound(hdld+1, hdld+num+1, L + a[i]) - hdld;
    int r = lower_bound(hdld+1, hdld+num+1, R + a[i]) - hdld - 1;
    int p = lower_bound(hdld+1, hdld+num+1, a[i]) - hdld;
    if (l <= r) {
        Segment res = query(1, l, r);
        if (res.mx) {
            f[i] = res.mx + 1, c[i] = res.cnt;
        } else {
            f[i] = 1, c[i] = 1;
        }
    } else {
        f[i] = 1, c[i] = 1;
    }
    ans = max(ans, f[i]);
    update(1, p, f[i], c[i]);
}

你以为这就结束了？

亲测：就算你该开 long long 的地方都开好了，还是会收获 听取WA一片。

为什么？因为 $c_i$ 可以非常大，甚至可以远超 $10^{18}$ 。

~~64 不够 128 来凑。（当然没必要。）~~

进一步分析，我们知道， $k$ 是相对比较小的。如果 $c_i$ 超过了 $k$ ，不管它超过了多少，都会被选掉。（看看上面贴出来的构造方案的片段，只要 $\le c_i$ ，不管差多少，都要进第二个分支。）

那么只要某个时候它超过了 $k$ ，手动给它赋值 $k + 1$ ，就不存在超 long long 表示范围的问题了。当然真正实现对 $c_i$ 做加法的是线段树里面，要在线段树里面修改。

代码

#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN = 5e5+5;

struct Segment {
    int l, r;
    int mx;
    long long cnt;
} s[MAXN<<2];

long long k, c[MAXN];
int n, L, R, a[MAXN], f[MAXN];
int hdld[MAXN], num, ans;

Segment merge(Segment lc, Segment rc) {
    Segment res;
    res.l = lc.l, res.r = rc.r;
    if (lc.mx > rc.mx) {
        res.mx = lc.mx, res.cnt = lc.cnt;
    } else if (lc.mx < rc.mx) {
        res.mx = rc.mx, res.cnt = rc.cnt;
    } else {
        res.mx = lc.mx, res.cnt = lc.cnt + rc.cnt;
    }
    if (res.cnt > k) res.cnt = k + 1;
    return res;
}

void pushup(int cur) {
    s[cur] = merge(s[cur*2], s[cur*2+1]);
}

void build(int cur, int l, int r) {
    s[cur].l = l, s[cur].r = r;
    if (l == r) {
        s[cur].mx = 0;
        s[cur].cnt = 1;
        return;
    }
    int mid = (s[cur].l + s[cur].r) >> 1;
    build(cur*2, l, mid);
    build(cur*2+1, mid+1, r);
    pushup(cur);
}

void update(int cur, int p, int mx, long long cnt) {
    if (s[cur].l == s[cur].r) {
        if (mx > s[cur].mx) {
            s[cur].mx = mx;
            s[cur].cnt = cnt;
        } else if (mx == s[cur].mx) {
            s[cur].cnt += cnt;
        }
        if (s[cur].cnt > k) s[cur].cnt = k + 1;
        return;
    }
    int mid = (s[cur].l + s[cur].r) >> 1;
    if (p <= mid) update(cur*2, p, mx, cnt);
    else update(cur*2+1, p, mx, cnt);
    pushup(cur);
    if (s[cur].cnt > 1e18) {
        printf("%lld\n", s[cur].cnt);
        exit(0);
    }
}

Segment query(int cur, int l, int r) {
    if (l <= s[cur].l && r >= s[cur].r) {
        return s[cur];
    }
    int mid = (s[cur].l + s[cur].r) >> 1;
    if (r <= mid) return query(cur*2, l, r);
    if (l > mid) return query(cur*2+1, l, r);
    return merge(query(cur*2, l, r), query(cur*2+1, l, r));
}

int main() {
    #ifndef ONLINE_JUDGE
    freopen("ibvl.in", "r", stdin);
    freopen("ibvl.out", "w", stdout);
    #endif
    scanf("%d%lld%d%d", &n, &k, &L, &R);
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &a[i]);
        hdld[i] = a[i];
    }
    sort(hdld+1, hdld+n+1);
    num = unique(hdld+1, hdld+n+1) - hdld - 1;
    build(1, 1, num);
    for (int i = n; i >= 1; --i) {
        int l = upper_bound(hdld+1, hdld+num+1, L + a[i]) - hdld;
        int r = lower_bound(hdld+1, hdld+num+1, R + a[i]) - hdld - 1;
        int p = lower_bound(hdld+1, hdld+num+1, a[i]) - hdld;
        if (l <= r) {
            Segment res = query(1, l, r);
            if (res.mx) {
                f[i] = res.mx + 1, c[i] = res.cnt;
            } else {
                f[i] = 1, c[i] = 1;
            }
        } else {
            f[i] = 1, c[i] = 1;
        }
        ans = max(ans, f[i]);
        update(1, p, f[i], c[i]);
    }
    printf("%d\n", ans);
    for (int i = 1, lst = 0; i <= n && ans; ++i) {
        if (f[i] == ans && (!lst || (L+a[lst]<a[i] && R+a[lst]>a[i]))) {
            if (k > c[i]) k -= c[i];
            else {
                printf("%d ", i);
                lst = i;
                --ans;
            }
        }
    }
    return 0;
}

winsunboy

关注

7
点赞
踩
14

收藏

觉得还不错? 一键收藏
0
评论
Luogu 5010 HMR的LIS Ⅲ 题解

给定长度为n的序列a1a2an，以及常数LR。要求构造序列b，满足（设b序列的长度为m∀1≤i≤m1≤bi≤n∀1≤imbibi1∀1≤imLabi1−abiR并在此基础上使得长度m最大。然而，在保证长度最大的前提下，仍然有许多中可能的构造方案。给定k，输出字典序第k小的构造方案。对于100%1≤n≤5×1051≤k≤10130≤∣L∣∣。
复制链接

扫一扫