[ABC339E] Smooth Subsequence 题解

最新推荐文章于 2024-07-20 09:21:06 发布

zhujiangyuan2027

最新推荐文章于 2024-07-20 09:21:06 发布

阅读量552

点赞数 12

分类专栏： AtCoder 题解文章标签：算法 c++

本文链接：https://blog.csdn.net/zhujiangyuan2027/article/details/137840603

版权

题解同时被 2 个专栏收录

18 篇文章 0 订阅

订阅专栏

AtCoder

14 篇文章 0 订阅

订阅专栏

文章讲述了如何使用线段树优化动态规划算法，解决最大子序列问题，将时间复杂度从O(n^2)降低到O(nlogV)，以求解以特定范围内的整数结尾的子序列的最大长度。

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

文章目录

题目传送门

首先很快想到一个 $O(n^2)$ 的 DP：

状态：设 $dp_i$ 为以 $a_i$ 为结尾的子序列的最大长度。

转移：若 $\left|a_i-a_j\le d\right|$ ， $dp_i=\max (dp_i,dp_j+1)$ 。

答案： $\max\limits_{i=1}^{n}dp_i$ 。

但是 $O(n^2)$ 的复杂度接受不了，考虑线段树优化 DP。使用权值线段树，下标为 $a_i$ 的值。查找 $a_i - d \sim a_i + d$ 之间 $d p$ 的最大值，然后更新即可。

时间复杂度 $O(n\log V)$ ， $V$ 为值域。

// by zhujiangyuan 2024/4/16
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 5e5;
int n, d, a[N + 10];
struct Seg {
	int dat;
	#define dat(p) t[p].dat
	#define ls p << 1
	#define rs p << 1 | 1
}t[N * 4];
int query (int p, int l, int r, int L, int R) {
	if (L <= l && r <= R) return dat(p);
	int mid = (l + r) >> 1, res = 0;
	if (L <= mid) res = max (res, query (ls, l, mid, L, R));
	if (mid < R) res = max (res, query (rs, mid + 1, r, L, R));
	return res;
}
void modify (int p, int l, int r, int pos, int val) {
	if (l == r) { dat(p) = max (dat(p), val); return ;}
	int mid = (l + r) >> 1;
	if (pos <= mid) modify (ls, l, mid, pos, val);
	else modify (rs, mid + 1, r, pos, val);
	dat(p) = max (dat(ls), dat(rs));
}
int main()
{
	scanf("%d %d", &n, &d);
	for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
	for (int i = n; i >= 1; i--) {
		int maxx = query (1, 1, N, max (a[i] - d, 1), min (a[i] + d, N));
		modify (1, 1, N, a[i], maxx + 1);
	}
	printf("%d", query(1, 1, N, 1, N));
	return 0;
}