HDU3450 Counting Sequences（树状数组+dp）

题意：找出长度大于1的子序列且每一对相邻的数差不超过d的个数。

可以很容易想到n^2的算法，蓝儿n有1e5，所以想到了dp。

用bit数组存储符合条件的序列的个数，bit[i]表示以i结尾的符合条件的个数。从头开始检索。对每个a[i]，怎么计算符合条件的个数呢？，由于我们的数组记录的是以a[i]结尾的，由题可知，我们要找的与a[i]相邻的数t要满足a[i]-d<=t<=a[i]+d。那么只需要求出sum(a[i]+d)和sum(a[i]-d)，就可以更新数组了。当然a[i]-d和a[i]+d是不一定存在的，所以要找出最小的大于等于a[i]-d的和最大的小于等于a[i]+d的。

怎么高效的更新sum呢？很容易想到树状数组。将原a数组去重排序得到新的b数组，那么sum函数里的参数就换为该数在b中的下标。由于是有序的，所以可以二分查找在b数组中的下标，更新sum也是logn的，所以总复杂度为O（n（logn+logn））= O（nlogn）。

代码：

// Header.
#include <algorithm>
#include <iostream>
#include <sstream>
#include <cstring>
#include <cstdio>
#include <vector>
#include <string>
#include <bitset>
#include <queue>
#include <stack>
#include <cmath>
#include <ctime>
#include <set>
#include <map>
using namespace std;

// Macro
typedef long long LL;
#define TIME cerr << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << "s." << endl;
#define IN freopen("/Users/apple/input.txt", "r", stdin);
#define OUT freopen("/Users/apple/out.txt", "w", stdout);
#define mem(a, n) memset(a, n, sizeof(a))
#define rep(i, n) for(int i = 0; i < (n); i ++)
#define REP(i, t, n) for(int i = (t); i < (n); i ++)
#define FOR(i, t, n) for(int i = (t); i <= (n); i ++)
#define ALL(v) v.begin(), v.end()
#define Min(a, b) a = min(a, b)
#define Max(a, b) a = max(a, b)
#define put(a) printf("%d\n", a)
#define ss(a) scanf("%s", a)
#define si(a) scanf("%d", &a)
#define sii(a, b) scanf("%d%d", &a, &b)
#define siii(a, b, c) scanf("%d%d%d", &a, &b, &c)
#define VI vector<int>
#define pb push_back
const int inf = 0x3f3f3f3f, N = 1e5 + 5, MOD = 9901;
// Macro end

int T, cas = 0;
int n, d;
int a[N], b[N], dp[N];
int bit[N];

// Imp
void add(int i, int x) {
    while(i < N) {
        bit[i] = (bit[i] + x) % MmOD;
        i += i & (-i);
    }
}
int sum(int x) {
    int ret = 0;
    while(x > 0) {
        ret = (ret + bit[x]) % MOD;
        x -= x & (-x);
    }
    return ret;
}


int main(){
#ifdef LOCAL
    IN // OUT
#endif
    
    while(sii(n, d) != EOF) {
        int idx = 0;
        mem(bit, 0);
        rep(i, n) {
            si(a[i]);
            b[i] = a[i];
        }
        sort(b, b + n);
        dp[idx++] = b[0];
        REP(i, 1, n) if(b[i] != b[i-1]) dp[idx++] = b[i];
        dp[idx++] = inf;
        rep(i, n) {
            int x = lower_bound(dp, dp + idx, a[i]) - dp + 1;
            int l = lower_bound(dp, dp + idx, a[i] - d) - dp;
            int r = upper_bound(dp, dp + idx, a[i] + d) - dp;
            int tmp = ((sum(r) - sum(l) + 1) % MOD + MOD) % MOD;
            add(x, tmp);
        }
        int ans = ((sum(idx) - n) % MOD + MOD) % MOD;
        put(ans);
    }
    
    return 0;
}