【线段树】第十三届蓝桥杯省赛C++ A组 Java C组 Python A组/B组《最长不下降子序列》(C++)

【题目描述】

给定一个长度为 N 的整数序列:,,⋅⋅⋅,

现在你有一次机会,将其中连续的 K 个数修改成任意一个相同值。

请你计算如何修改可以使修改后的数列的最长不下降子序列最长,请输出这个最长的长度。

最长不下降子序列是指序列中的一个子序列,子序列中的每个数不小于在它之前的数。

【输入格式】

输入第一行包含两个整数 N 和 K。

第二行包含 N 个整数 ,,⋅⋅⋅,

【输出格式】

输出一行包含一个整数表示答案。

【数据范围】

对于 20% 的评测用例,1≤K≤N≤100;

对于 30% 的评测用例,1≤K≤N≤1000;

对于 50% 的评测用例,1≤K≤N≤10000;

对于所有评测用例,1 ≤ K ≤ N ≤ ,1≤

【输入样例】

5 1
1 4 2 8 5

【输出样例】

4

【思路】

题解来源:AcWing 4648. 最长不下降子序列 - AcWing

【代码】

cpp 复制代码
#include <bits/stdc++.h>
const int N = 1e5 + 10;
int n, m, k, ans;
int a[N];
int b[N];       // 用于离散化的数组
int dp1[N];     // dp1[i]表示从前往后以a[i]结尾的最长不下降子序列的长度
int dp2[N];     // dp2[i]表示从前往后以a[i]开头的最长不下降子序列的长度
int find(int x) //返回整数a[i]在b数组中的下标
{
    int l = 1, r = m;
    while (l < r)
    {
        int mid = l + r >> 1;
        if (b[mid] >= x)
            r = mid;
        else
            l = mid + 1;
    }
    return l;
}
struct
{
    int maxv;
} seg[N * 4];
void pushup(int id)
{
    seg[id].maxv = std::max(seg[id << 1].maxv, seg[id << 1 | 1].maxv);
}
void build(int id, int l, int r)
{
    if (l == r)
        seg[id].maxv = 0;
    else
    {
        int mid = l + r >> 1;
        build(id << 1, l, mid);
        build(id << 1 | 1, mid + 1, r);
        pushup(id);
    }
}
void change(int id, int l, int r, int pos, int val)
{
    if (l == r)
        seg[id].maxv = std::max(seg[id].maxv, val);
    else
    {
        int mid = l + r >> 1;
        if (pos <= mid)
            change(id << 1, l, mid, pos, val);
        else
            change(id << 1 | 1, mid + 1, r, pos, val);
        pushup(id);
    }
}
int query(int id, int l, int r, int ql, int qr)
{
    if (l == ql && r == qr)
        return seg[id].maxv;
    int mid = l + r >> 1;
    if (qr <= mid)
        return query(id << 1, l, mid, ql, qr);
    else if (ql >= mid + 1)
        return query(id << 1 | 1, mid + 1, r, ql, qr);
    else
        return std::max(query(id << 1, l, mid, ql, mid), query(id << 1 | 1, mid + 1, r, mid + 1, qr));
}
signed main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);
    std::cout.tie(0);
    std::cin >> n >> k;
    for (int i = 1; i <= n; ++i)
        std::cin >> a[i], b[i] = a[i];
    std::sort(b + 1, b + n + 1); //排序
    if (n == k)
    {
        std::cout << n << '\n';
        return 0;
    } 
    m = 1;
    for (int i = 2; i <= n; i++) //去重
        if (b[i] != b[m])
            b[++m] = b[i];
    for (int i = 1; i <= n; ++i)
        a[i] = find(a[i]);
    build(1, 1, m); //建权值线段树
    for (int i = 1; i <= n - k; ++i)
    {
        dp1[i] = query(1, 1, m, 1, a[i]) + 1;
        change(1, 1, m, a[i], dp1[i]);
    }
    build(1, 1, m); // dp1已经处理完,重新建树处理dp2
    for (int i = n; i >= k + 1; --i)
    {
        ans = std::max(ans, dp1[i - k] + k + query(1, 1, m, a[i - k], m));
        // 第一段为dp1[i-k],第二段为k,第三段为max(dp2[j]),i+k+1<=j<=n且a[j]>=a[i]
        dp2[i] = query(1, 1, m, a[i], m) + 1;
        change(1, 1, m, a[i], dp2[i]);
    }
    // 特殊情况
    for (int i = 1; i <= n - k; ++i)
        ans = std::max(ans, dp1[i] + k);
    for (int i = n; i >= k + 1; --i)
        ans = std::max(ans, dp2[i] + k);
    std::cout << ans << '\n';
    return 0;
}
相关推荐
passer__jw76718 分钟前
【LeetCode】【算法】283. 移动零
数据结构·算法·leetcode
Ocean☾25 分钟前
前端基础-html-注册界面
前端·算法·html
顶呱呱程序33 分钟前
2-143 基于matlab-GUI的脉冲响应不变法实现音频滤波功能
算法·matlab·音视频·matlab-gui·音频滤波·脉冲响应不变法
爱吃生蚝的于勒1 小时前
深入学习指针(5)!!!!!!!!!!!!!!!
c语言·开发语言·数据结构·学习·计算机网络·算法
羊小猪~~1 小时前
数据结构C语言描述2(图文结合)--有头单链表,无头单链表(两种方法),链表反转、有序链表构建、排序等操作,考研可看
c语言·数据结构·c++·考研·算法·链表·visual studio
王哈哈^_^1 小时前
【数据集】【YOLO】【VOC】目标检测数据集,查找数据集,yolo目标检测算法详细实战训练步骤!
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·pyqt
星沁城1 小时前
240. 搜索二维矩阵 II
java·线性代数·算法·leetcode·矩阵
脉牛杂德2 小时前
多项式加法——C语言
数据结构·c++·算法
legend_jz2 小时前
STL--哈希
c++·算法·哈希算法
CSUC2 小时前
【C++】父类参数有默认值时子类构造函数列表中可以省略该参数
c++