[HDU - 2795] 小明序列 (线段树或DP)

链接

http://acm.hdu.edu.cn/showproblem.php?pid=4521

题意

给定一个序列a，求间隔大于等于d的最长上升子序列，输出这个长度；

分析

可以用DP；
这里使用树状数组，首先根据序列中最大的数字建立一个线段树，其中线段树的每个节点[l,r]都有一个max，存储[l,r]中间隔大于等于d的最长上升子序列，其中叶子节点[i,i]代表，以i结尾的满足条件的子序列的最大长度，叶子节点显然最大为1，初始的时候全部为0，我们还需要一个数组dp[i]记录以i结尾间隔大于等于d的最长上升子序列，当下标小于等于d+1时，显然dp[1]全为1，因为间隔大于等于d，在此之前我们不做插入操作，当第d+1个数之后时，我们插入在a[i-d-1]，即与第i个数相距d的前面那个数的叶子位置插入dp[i-d-1]（dp[i-d-1]一定是我们之前求出来的），之后我们求出区间[1,a[i]-1]的max，即以a[i]-1结尾的满足条件的最长上升子序列，然后dp[i]=max+1，进行更新即可；关于间距，因为我们只插入了i-d-1之前的数据，所以保证了间距，如果a[i]为0，就令dp[i]=0；

代码

#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <set>
#include <map>

using namespace std;

#define inf 0x7f7f7f7f
#define maxn 100005
#define N 200005
#define P 2
typedef long long ll;

int t, n, d, a[maxn];

struct Tree {
    int l, r, max;
} tree[4 * maxn];

void push_up(int x) {
    tree[x].max = max(tree[x << 1].max, tree[x << 1 | 1].max);
}

void build(int x, int l, int r) {
    tree[x].l = l, tree[x].r = r;
    if (l == r) {
        tree[x].max = 0;
        return;
    }
    int mid = (l + r) >> 1;
    build(x << 1, l, mid);
    build(x << 1 | 1, mid + 1, r);
    push_up(x);
}

void update(int x, int pos, int val) {
    int l = tree[x].l, r = tree[x].r;
    if (l == r) {
        tree[x].max = max(tree[x].max, val);
        return;
    }
    int mid = (l + r) >> 1;
    if (pos <= mid)update(x << 1, pos, val);
    else update(x << 1 | 1, pos, val);
    push_up(x);
}

int query(int x, int l, int r) {
    int le = tree[x].l, ri = tree[x].r;
    if (l <= le && ri <= r) {
        return tree[x].max;
    }
    int mid = (le + ri) >> 1;
    int maxm = 0;
    if (l <= mid) maxm = max(maxm, query(x << 1, l, r));
    if (r > mid) maxm = max(maxm, query(x << 1 | 1, l, r));
    return maxm;
}

int dp[maxn];

int main() {
    while (cin >> n >> d) {
        int maxm = 0;
        for (int i = 1; i <= n; i++) {
            cin >> a[i];
            maxm = max(a[i], maxm);
        }
        memset(dp, 0, sizeof(dp));
        build(1, 0, maxm);
        int res = 0;
        for (int i = 1; i <= n; i++) {
            if (i - d - 1 >= 1)update(1, a[i - d - 1], dp[i - d - 1]);
            if (a[i] == 0)dp[i] = 1;
            else dp[i] = query(1, 0, a[i] - 1) + 1;
            res = max(dp[i], res);
        }
        cout << res << endl;
    }
}