【线段树优化dp】zoj3349 Special Subsequence

题目点这里

题意：在一个序列中寻找一个最长字序列使得每个相邻的数绝对值相差不差过d

最开始以为是子数组 那这道题就水了 结果是子序列 然后就orz了

dp[i] 表示包含前i个数的子序列中 最长的符合要求的子序列的长度 dp[i] = max(dp[j] + 1) ( j < i && abs(a[i] - a[j]) <= d )

用权值线段树维护dp的最大值 每次查询区间 a[i] - d 到 a[i] + d 的最大值

由于ai的范围比较大需要离散化（还好不是恶心的强制在线orz） 用stl就可以了

先sort排序 然后unique去重 lower_bound(a, a + len, x) - a返回数组a中的x离散化后的值

（lower_bound返回第一个不大于查询值的指针 upper_bound返回第一个不小于查询值的指针）

最开始M和N搞混了一直TLE orzzzzzzzzz

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>

#define lc u << 1
#define rc u << 1 | 1

using namespace std;

const int maxN = 100005;

int N, D;
int bfr[maxN], aft[maxN];

struct BIT{
	int mmax[3 * maxN];
	
	inline void init() { memset(mmax, 0, sizeof(mmax)); }
	
	int query(int u, int l, int r, int L, int R)
	{
		if(l == L && r == R) return mmax[u];
		int mid = (l + r) >> 1;
		if(R <= mid) return query(lc, l, mid, L, R);
		else if(L > mid) return query(rc, mid + 1, r, L, R);
		else return max( query(lc, l, mid, L, mid), query(rc, mid + 1, r, mid + 1, R) ); 
	}
	
	void add(int u, int l, int r, int pos, int w)
	{
		if(l == r){ mmax[u] = w; return; } 
		int mid = (l + r) >> 1;
		if(pos <= mid) add(lc, l, mid, pos, w);
		else add(rc, mid + 1, r, pos, w);
		mmax[u] = max(mmax[lc], mmax[rc]);
	}
}t;

int main()
{
	while(~scanf("%d%d", &N, &D))
	{
		for(int i = 0; i < N; ++i)
		{
			scanf("%d", bfr + i);
			aft[i] = bfr[i];
		}
		sort(aft, aft + N);
		int M = unique(aft, aft + N) - aft;
		
		int ans = 0; t.init();
		for(int i = 0; i < N; ++i)
		{
			int L = lower_bound(aft, aft + M, bfr[i] - D) - aft;
			int R = upper_bound(aft, aft + M, bfr[i] + D) - aft - 1;
			int cnt = t.query(1, 0, M - 1, L, R) + 1;
			ans = max(ans, cnt);
			t.add(1, 0, M - 1, lower_bound(aft, aft + M, bfr[i]) - aft, cnt);
		}
		printf("%d\n", ans);
	}
	
	return 0;
}