Atcoder-abl_d-Flat Subsequence题解（线段树+dp）

input#00ff00

已于 2023-02-25 14:24:10 修改

阅读量193

点赞数

文章标签： c++ 算法数据结构动态规划 Powered by 金山文档

于 2023-02-04 17:51:33 首次发布

本文链接：https://blog.csdn.net/qq_42987482/article/details/128883745

版权

文章介绍了如何解决一个关于找到最长子序列的问题，其中序列中最小值和最大值的差不超过给定的整数K。通过动态规划和线段树的数据结构优化，可以有效地在O(nlogn)的时间复杂度内找到答案。代码示例展示了线段树的构建和更新过程，以及如何使用它来查询和更新最大值。

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

题目：

题目链接：https://atcoder.jp/contests/abl/tasks/abl_d

题意，给定n和k，求最长子序列使得序列内最小值和最大值的相差不超过k，输出序列的长度。

思路：

注意题中要求的子序列可以不连续，namo，值与值之间互相限制同时又有递推、继承的关系，朴素的思路就是dp，dp[i]表示将第i个数选入序列中后，能构成的最长序列，不难发现，dp[i]只受到相差在k以内的元素的约束，取最大即可，所以可以得到转移关系：

dp[i]=max(dp[j])+1//abs(a[i]-a[j])<=k

我们可以用二重循环去找所有满足条件的j，但这种做法的时间复杂度是 $\text{[math]}$ ，但是此处可以用线段树的数据结构优化：要做的是对于所有的j满足 $\text{[math]}$ ，找最大的dp[j]。

Namo，他就是一个偏序问题，两个属性，一个a一个dp值，在限制a的大小的情况下求最大的dp值，那可以用线段树来求解，也就是找区间最大值，将数值对应到下标，dp值对应到权值，构造权值线段树即可解决此题。

小细节：

1.注意a[i]-k和a[i]+k可能会超过数据范围，导致线段树报错，询问的左右端点要进行预处理。

2.这里的upd函数要做的是更新节点为最大值，即：

    if (l == r)
    {
        t[p] =max(t[p],x);
        return;
    }

AC code:

#include<iostream>
#include<cmath>
#include<algorithm>
#include<map>
#include<vector>
#include<cstring>
#include<set>
#include<queue>
#define eol "\n"
typedef long long ll;
typedef unsigned long long ull;
const int inf = 0x3f3f3f3f;
const int mod = 998244353;
const int N   = 2E6 + 7;
using namespace std;
ll ksm(ll a,ll b) {ll res=1;a%=mod; while(b){if(b&1)res=res*a%mod;b>>=1;a=a*a%mod;}return res%mod;}
ll gcd(ll a,ll b) {return b==0?a:gcd(b,a%b);}
ll lcm(ll a,ll b) {return a/gcd(a,b)*b;}
ll inv(ll x) {return ksm(x, mod-2);}
int t[N << 2];
void up(int p) { t[p] = max(t[p << 1], t[p << 1 | 1]); }
void update(int p, int l, int r,int f, int x){
    if (l == r)
    {
        t[p] =max(t[p],x);
        return;
    }
    int mid = l + r >> 1;
    if (f<=mid)update(p << 1, l, mid, f, x);
    else if(f>mid) update(p << 1|1, mid+1, r, f, x);
    up(p);
}
//区间查询（查询区间a，b的最大值）
int xquery(int p,int l,int r,int a,int b) {
    int ans = -1;
    if (a <= l && r <= b) {
        return t[p];
    }
    int mid = (l + r) / 2;
    if (a <= mid)ans = max(ans, xquery(p << 1, l, mid, a, b));
    if (b > mid)ans = max(ans, xquery(p << 1 | 1, mid + 1, r, a, b));
    return ans;
}
ll dp[N];
void solve(){
    int n,k;cin>>n>>k;
    for(int i=1;i<=n;i++){
        int temp;cin>>temp;
        int le=max(temp-k,0),r=min(300001,temp+k);
        if(le<=r)
            dp[i]= xquery(1,0,300001,le,r)+1;
        update(1,0,300001,temp,dp[i]);
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
        ans=max(ans,dp[i]);
    cout<<ans<<eol;
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    int t=1;
    //cin>>t;
    while(t--)
        solve();
}